numpy.swapaxes

numpy.swapaxes(a, axis1, axis2)

交换数组的两个轴.

参数:

aarray_like

输入数组.

axis1int

第一轴.

axis2int

第二轴.

返回:

a_swappedndarray

对于 NumPy >= 1.10.0,如果 a 是一个 ndarray,则返回 a 的一个视图;否则创建一个新的数组.对于更早的 NumPy 版本,只有当轴的顺序改变时,才返回 a 的一个视图,否则返回输入数组.

示例

>>> import numpy as np
>>> 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]]])