python - 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 中是相反的。