numpy - 二维数组索引

Question

According to docs numpy's default behaviour is to index arrays first by rows then by columns:

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

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

print a[0][1] # is 1

I want to index the array using the geometrically oriented-convention a[x][y] , as in x-axis and y-axis. How can I change the indexing order without modifying the array's shape so that a[0][1] would return 2?

Answer 1

您可以编写aT[0,1]来使用数组转置的索引，这在2D中是相反的。