如何使用python返回数组中的最大值索引？

Question

我有一个数组，我想找到最大值的索引。

例如：

myarray = np.array([1,8,8,3,2])

我想得到结果：[1,2]，我该怎么做？

（实际上我尝试了np.argmax(myarray)，但它只返回第一次出现[1]）

Answer 1

假设：

>>> myarray = np.array([1,8,8,3,2])

你可以这样做：

>>> np.where(myarray==myarray[np.argmax(myarray)])
(array([1, 2]),)

或，

>>> np.where(myarray==max(myarray))
(array([1, 2]),)

或，

>>> np.nonzero(myarray==max(myarray))
(array([1, 2]),)