Python numpy stack用法及代碼示例

本文簡要介紹 python 語言中 numpy.stack 的用法。

用法:

numpy.stack(arrays, axis=0, out=None)

沿新軸加入一係列數組。

axis 參數指定新軸在結果維度中的索引。例如，如果 axis=0 它將是第一個維度，如果 axis=-1 它將是最後一個維度。

參數：

arrays： 數組 序列

每個數組必須具有相同的形狀。

axis： 整數，可選

結果數組中輸入數組沿其堆疊的軸。

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