Python numpy swapaxes用法及代碼示例

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

用法:

numpy.swapaxes(a, axis1, axis2)

交換數組的兩個軸。

參數：

a： array_like

輸入數組。

axis1： int

第一個軸。

axis2： int

第二軸。

返回：

a_swapped： ndarray

對於NumPy >= 1.10.0，如果a是ndarray，則返回a的視圖；否則將創建一個新數組。對於早期的 NumPy 版本，僅當軸的順序更改時才返回 a 的視圖，否則返回輸入數組。

例子：

>>> x = np.array([[1,2,3]])
>>> np.swapaxes(x,0,1)
array([[1],
       [2],
       [3]])

>>> x = np.array([[[0,1],[2,3]],[[4,5],[6,7]]])
>>> x
array([[[0, 1],
        [2, 3]],
       [[4, 5],
        [6, 7]]])

>>> np.swapaxes(x,0,2)
array([[[0, 4],
        [2, 6]],
       [[1, 5],
        [3, 7]]])