Функция swapaxes

Функция swapaxes меняет местами две указанные оси многомерного массива NumPy. Первым параметром функция принимает массив, вторым - номер первой оси, третьим - номер второй оси. Возвращается новый массив с переставленными осями, исходный массив при этом не изменяется.

Синтаксис

np.swapaxes(arr, axis1, axis2)

Пример

Давайте создадим двумерный массив и поменяем местами его оси:

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
res = np.swapaxes(arr, 0, 1)
print(res)

Результат выполнения кода:

[[1 4]
 [2 5]
 [3 6]]

Пример

Давайте создадим трехмерный массив и поменяем местами оси с индексом 0 и 2:

import numpy as np

arr = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
res = np.swapaxes(arr, 0, 2)
print(res)

Результат выполнения кода:

[[[1 5]
  [3 7]]

 [[2 6]
  [4 8]]]

Пример

Давайте используем метод swapaxes у объекта массива:

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
res = arr.swapaxes(0, 1)
print(res)

Результат выполнения кода:

[[1 4]
 [2 5]
 [3 6]]

Пример

Давайте посмотрим на форму массива до и после перестановки осей:

import numpy as np

arr = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
print(arr.shape)
res = np.swapaxes(arr, 0, 2)
print(res.shape)

Результат выполнения кода:

(2, 2, 2)
(2, 2, 2)

Смотрите также

функцию transpose,
которая переставляет оси массива в произвольном порядке
функцию moveaxis,
которая перемещает указанную ось на новую позицию
функцию reshape,
которая изменяет форму массива без изменения данных
функцию expand_dims,
которая добавляет новую ось в массив