python - 返回非平面索引的 numpy 数组的 Argmax

Question

我正在尝试获取 Numpy 数组中最大元素的索引。这可以使用 numpy.argmax 来完成。我的问题是，我想找到整个数组中最大的元素并获得它的索引。

numpy.argmax 可以沿一个轴应用，这不是我想要的，也可以在扁平数组上应用，这正是我想要的。

我的问题是，当我想要多维索引时，将 numpy.argmax 与 axis=None 一起使用会返回平面索引。

我可以使用 divmod 来获得非平面索引，但这感觉很难看。有没有更好的方法呢？

Answer 1

您可以使用 numpy.unravel_index()关于 numpy.argmax() 的结果:

>>> a = numpy.random.random((10, 10))
>>> numpy.unravel_index(a.argmax(), a.shape)
(6, 7)
>>> a[6, 7] == a.max()
True