numpy.stack

numpy.stack(arrays, axis=0)

沿着一个新的轴连接一系列数组。

轴参数指定结果维度中新轴的索引。例如，如果axis=0，它将是第一个维度，如果axis=-1，它将是最后一个维度。

版本1.10.0新增。

参数：

arrays：array_like 序列

每个数组必须具有相同的形状。

axis：int，可选

结果数组中的轴，输入数组将沿其连接。

返回：

stacked：ndarray

连接后的数组比输入数组多一个维度。

另见

concatenate

沿一个已存在的轴连接一系列阵列。

split

将数组拆分成大小相同的多个子数组列表。

例子

>>> arrays = [np.random.randn(3, 4) for _ in range(10)]
>>> np.stack(arrays, axis=0).shape
(10, 3, 4)

>>> np.stack(arrays, axis=1).shape
(3, 10, 4)

>>> np.stack(arrays, axis=2).shape
(3, 4, 10)

>>> a = np.array([1, 2, 3])
>>> b = np.array([2, 3, 4])
>>> np.stack((a, b))
array([[1, 2, 3],
       [2, 3, 4]])

>>> np.stack((a, b), axis=-1)
array([[1, 2],
       [2, 3],
       [3, 4]])