numpy stack 用法及代码示例

用法:
numpy.stack(arrays, axis=0, out=None)
沿新轴连接一 Series 数组。

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

1.10.0版中的新功能。

描述
参数 arrays: : sequence of array_like

每个数组必须具有相同的形状。
axis: : int, 可选参数

结果数组中输入轴沿其堆叠的轴。
out: : ndarray, 可选参数

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

返回值: stacked: : ndarray

堆叠的数组比输入数组多一维。

例子:

>>> 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]])

发表回复

登录... 后才能评论