numpy.permute_dims

numpy.permute_dims(a, axes=None)[source]

Возвращает массив с переставленными осями.

Для одномерного массива возвращает неизменную ссылку на исходный массив, поскольку переставленный вектор — это просто тот же вектор. Чтобы преобразовать одномерный массив в двумерный столбец, необходимо добавить дополнительное измерение, например, np.atleast_2d(a).T достигает этого, как и a[:, np.newaxis]. Для двумерного массива это стандартная транспонирование матрицы. Для n-мерного массива, если заданы оси, их порядок указывает, как переставляются оси (см. примеры). Если оси не заданы, то transpose(a).shape == a.shape[::-1].

Параметры:

aarray_like

Входной массив.

axesкортеж или список целых чисел, необязательно

Если указано, это должен быть кортеж или список, содержащий перестановку [0,1,…,N-1], где N — количество осей a. i-я ось возвращаемого массива будет соответствовать оси с номером axes[i] входного массива. Если не указано, по умолчанию range(a.ndim)[::-1], что инвертирует порядок осей.

Возвращает:

pndarray

a с переставленными осями. Возвращается ссылка на представление, когда это возможно.

См. также

ndarray.transpose

Эквивалентный метод.

moveaxis

Переместить оси массива в новые позиции.

argsort

Возвращает индексы, которые отсортируют массив.

Примечания

Используйте transpose(a, argsort(axes)) для инвертирования транспонирования тензоров при использовании ключевого аргумента axes.

Примеры

>>> a = np.array([[1, 2], [3, 4]])
>>> a
array([[1, 2],
       [3, 4]])
>>> np.transpose(a)
array([[1, 3],
       [2, 4]])

>>> a = np.array([1, 2, 3, 4])
>>> a
array([1, 2, 3, 4])
>>> np.transpose(a)
array([1, 2, 3, 4])

>>> a = np.ones((1, 2, 3))
>>> np.transpose(a, (1, 0, 2)).shape
(2, 1, 3)

>>> a = np.ones((2, 3, 4, 5))
>>> np.transpose(a).shape
(5, 4, 3, 2)