How to get all indices for the maximum value in a numpy array?

Question

I have an array of numbers, and the maximum value might occurrence more than once.

How can we get a collection of indices of all the occurrences of the maximum value in the array?

For example, for the following array:

import numpy as np

a = np.array((1,2,3,2,3,2,1,3))

the result should be [2, 4, 7] (or an equivalent array or tuple).

eumiro · Accepted Answer · 2011-08-12 10:51:35Z

11

import numpy as np

a = np.array((1,2,3,2,3,2,1,3))

occurences = np.where(a == a.max())

# occurences == array([2, 4, 7])

answered Aug 12, 2011 at 10:51

eumiro

214k36 gold badges307 silver badges264 bronze badges

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

1 Comment

dtlussier Over a year ago

I'm not sure if this is true with more recent versions, but with v1.4.0, should have occurences = np.where(a == a.max())[0] such that you get an array rather than a tuple in occurences.

Collectives™ on Stack Overflow

How to get all indices for the maximum value in a numpy array?

1 Answer 1

1 Comment

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Linked

Related