获取numpy数组中最大项的位置

Question

如何才能获得多维numpy数组中最大项的位置？

Answer 1

该argmax()方法应该有所帮助.

更新

(阅读评论后)我相信该argmax()方法也适用于多维数组.链接的文档提供了一个示例:

>>> a = array([[10,50,30],[60,20,40]])
>>> maxindex = a.argmax()
>>> maxindex
3

更新2

(感谢KennyTM的评论)您可以使用unravel_index(a.argmax(), a.shape)索引作为元组:

>>> from numpy import unravel_index
>>> unravel_index(a.argmax(), a.shape)
(1, 0)

Answer 2

（编辑）我指的是一个已经删除的旧答案。公认的答案是我的。我同意那argmax比我的答案更好。

这样做是否更具可读性/直观性？

numpy.nonzero(a.max() == a)
(array([1]), array([0]))

要么，

numpy.argwhere(a.max() == a)