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]使用数组转置的索引,这是2D中的另一种方式.