python numpy transpose函数

numpy transpose函数

>>> import numpy as np
>>> A = np.arange(16)
>>> A = A.reshape(2, 2, 4)
>>> print(A)
[[[ 0  1  2  3]
  [ 4  5  6  7]]

 [[ 8  9 10 11]
  [12 13 14 15]]]
>>> print(A.transpose(1, 0, 2))
[[[ 0  1  2  3]
  [ 8  9 10 11]]

 [[ 4  5  6  7]
  [12 13 14 15]]]
>>> 

A中的维度是(0, 1, 2)，假设对应着xyz
A.transpose(1, 0, 2)后，对应的维度则变为yxz
相当于将第一维和第二维进行了转置

举个例子：
数组A中11的坐标为(1，0，4)，A.transpose(1, 0, 2)后，11的坐标变为(0, 1, 4)。这个转置相当于将第一个坐标与第二坐标进行了互换。