swapaxes

swapaxes(a, axis1, axis2)

Interchange two axes of an array.

Parameters:

a : ndarray

Input array.

axis1 : int

First axis.

axis2 : int

Second axis.

Returns:

a_swapped : ndarray

Swapped array. This is always a copy of the input array.

Examples

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