Get the position of the largest value in a multi-dimensional NumPy array

(edit) I was referring to an old answer which had been deleted. And the accepted answer came after mine. I agree that argmax is better than my answer.

Wouldn't it be more readable/intuitive to do like this?

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

Or,

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

answered 2012-02-24T12:01:45.820

Get the position of the largest value in a multi-dimensional NumPy array

1 Answer

Your Answer