numpy.swapaxes

numpy.swapaxes(a, axis1, axis2) [source]

Меняет местами две оси массива.

Параметры:	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]]])