Python中的NumPy库是进行科学计算之前需要掌握的基础。在NumPy库当中,有一个非常实用的函数——stack()。它可以将多个数组沿着指定的轴拼接起来,生成一个新的数组。下面我们就来详细了解一下这个函数。
首先,我们需要导入NumPy库:
import numpy as np
然后我们创建几个数组,假设有两个数组a和b,它们的形状分别为(2,3)和(2,3):
a = np.array([[1, 2, 3], [4, 5, 6]])
b = np.array([[7, 8, 9], [10, 11, 12]])
现在我们来使用stack()函数将这两个数组沿着行的方向拼接起来:
c = np.stack((a, b), axis=0)
其中,第一个参数是要拼接的数组组成的元组,这里是(a, b);第二个参数是轴的方向,这里是0,表示沿着第一个维度(行)拼接。
运行结果如下:
array([[[ 1, 2, 3],
[ 4, 5, 6]],
[[ 7, 8, 9],
[10, 11, 12]]])
我们可以看到,拼接后得到了一个三维数组,其形状为(2,2,3),其中第一维表示拼接的数组个数,第二维和第三维表示原数组的形状。
除了按行拼接之外,我们还可以对列进行拼接。以数组a和b为例,我们可以将它们沿着列的方向拼接:
c = np.stack((a, b),