numpy stack 用法及代码示例
用法:
numpy.stack(arrays, axis=0, out=None)
沿新轴连接一 Series 数组。
的axis参数指定结果范围内新轴的索引。例如,如果axis=0这将是第一个维度,如果axis=-1这将是最后一个维度。
1.10.0版中的新功能。
– | 描述 |
---|---|
参数 | arrays: : sequence of array_like
每个数组必须具有相同的形状。 结果数组中输入轴沿其堆叠的轴。 如果提供,则为放置结果的目的地。形状必须正确,并且与未指定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]])
声明:本站所有文章,如无特殊说明或标注,均为本站原创发布。任何个人或组织,在未征得本站同意时,禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益,可联系我们进行处理。