numpy - 二维数组索引

Question

根据docs 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]来使用数组转置的索引，这是2D的另一种方式。