numpy.stack

numpy.stack(arrays, axis=0, out=None)[源代码]

沿新轴连接一系列数组。

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

1.10.0 新版功能.

参数
arrays阵列序列

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

axis可选的

结果数组中沿其堆叠输入数组的轴。

outndarray,可选

如果提供,则为放置结果的目的地。形状必须正确,与未指定out参数时堆栈将返回的形状匹配。

返回
stacked恩达雷

堆积数组比输入数组多了一个维度。

参见

concatenate

沿现有轴联接一系列数组。

block

从嵌套的块列表中组装nd数组。

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([4, 5, 6])
>>> np.stack((a, b))
array([[1, 2, 3],
       [4, 5, 6]])
>>> np.stack((a, b), axis=-1)
array([[1, 4],
       [2, 5],
       [3, 6]])