How to return the indices of maximum value from an array with python?

Question

I have an array and I want to find the indices of the maximum values.

For example:

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

I want to get the result: [1,2], how can I do that?

(Actually I tried np.argmax(myarray), but it only return the first occurrence [1])

Use max() to find the maximum value, then numpy.where() to find all the indexes with that value. — Barmar
– Barmar, Commented Sep 10, 2016 at 0:17
Bet there is a better way, but np.where(myarray==myarray[np.argmax(myarray)]) works... — dawg
– dawg, Commented Sep 10, 2016 at 0:17

dawg · Accepted Answer · 2016-09-10 00:27:09Z

3

Given:

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

You can do:

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

or,

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

or,

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

answered Sep 10, 2016 at 0:18

dawg

105k24 gold badges142 silver badges217 bronze badges

Sign up to request clarification or add additional context in comments.

1 Answer 1