numpy.swapaxes

numpy.swapaxes(a, axis1, axis2)

Функция swapaxes() меняет местами две оси массива.

Параметры:
a - массив Numpy
Массив, оси которого должны быть перемещены.
axis1 - целое число
Первая ось.
axis2 - целое число
Вторая ось.
Возвращает:
ndarray - массив NumPy
Массив у которого две оси поменялись позициями.
Смотрите так же: moveaxis, rollaxis, transpose

Замечание

С версии 1.10 возвращается представление, если на вход подается массив NumPy, в противном случае создается новый массив. Для более ранних версий представление возвращается только в том случае, если порядок осей изменяется, если нет, то возвращается исходный массив.



Примеры

>>> import numpy as np
>>> 
>>> a = np.arange(24).reshape(2, 3, 4)
>>> a
array([[[ 0,  1,  2,  3],
        [ 4,  5,  6,  7],
        [ 8,  9, 10, 11]],

       [[12, 13, 14, 15],
        [16, 17, 18, 19],
        [20, 21, 22, 23]]])
>>> 
>>> a.shape
(2, 3, 4)
>>> 
>>> b = np.swapaxes(a, 0, 2)
>>> b
array([[[ 0, 12],
        [ 4, 16],
        [ 8, 20]],

       [[ 1, 13],
        [ 5, 17],
        [ 9, 21]],

       [[ 2, 14],
        [ 6, 18],
        [10, 22]],

       [[ 3, 15],
        [ 7, 19],
        [11, 23]]])
>>> 
>>> b.shape
(4, 3, 2)
>>> 
>>> 
>>> a.shape
(2, 3, 4)
>>> 
>>> np.swapaxes(a, 0, 1).shape
(3, 2, 4)
>>> 
>>> np.swapaxes(a, 0, 2).shape
(4, 3, 2)
>>> 
>>> np.swapaxes(a, 2, 0).shape
(4, 3, 2)