Numpy:将一维中的索引转换为多维

Question

尽管数组是多维的,但许多数组方法返回单个索引.例如:

a = rand(2,3)
z = a.argmax()

对于二维,很容易找到最大元素的矩阵索引:

a[z/3, z%3]

但对于更多尺寸,它可能会变得烦人.在一个(折叠)维度中给定索引时,Numpy/Scipy是否有一种简单的方法可以在多个维度中返回索引？谢谢.

Answer 1

得到它了!

a = X.argmax()
(i,j) = unravel_index(a, X.shape)