numpy - 二维数组索引

Question

根据文档，numpy默认的数组索引方式是先按行再按列来访问数组：

a = numpy.arange(6).reshape(3,2)

[[0 1]
 [2 3]
 [4 5]]

print a[0][1] # is 1

我想用一种几何坐标的方式来索引数组，也就是用a[x][y]的形式，像是x轴和y轴那样。请问我怎么能改变索引的顺序，而不改变数组的形状，这样a[0][1]就能返回2呢？

Answer 1

你可以用 a.T[0,1] 来访问数组的转置，转置后的数组在二维情况下，行和列的位置是反过来的。