数组的转置和轴的变换

668 阅读1分钟

二维数组的转置十分简单 用一个array.T 就可以了

高维数组的转置则涉及了一个轴的变换问题 高维数组是比较难理解的概念,我上网查询的时候发现三维数组的理解,好几个人给出完全不同的答案,下面我画了个图,假设那是我的三维数组结构的认识

array = np.arange(16).reshape((2,2,4))
print(array)

运行结果是

[[[ 0  1  2  3]
  [ 4  5  6  7]]
 [[ 8  9 10 11]
  [12 13 14 15]]]

我这里画了一个图,分别用0,1,2代表x,y,z轴

我运行

print(array[0,1,3])
print(array[1,1,2])

得到

7
14

然后用transpose转置

array = array.transpose((1,0,2))
print(array)

然后结果是什么呢,我的理解是transpose后面括号默认的参数(0,1,2),分别代表x,y,z轴,如果改成(1,0,2)不就是x,y改变位置吗,于是,我画出图

运行输出结果

[[[ 0  1  2  3]
  [ 8  9 10 11]]
 [[ 4  5  6  7]
  [12 13 14 15]]]

检验一下

a = array[0,1,3]
print(a)

如果得到11就证明我的假设成立了 运行结果