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