Python: получить позицию самого большого элемента в массиве numpy

Question

Как я могу получить позицию самого большого элемента в многомерном массиве numpy?

Answer 1

Метод argmax() должен помочь.

Update

(После прочтения комментария) Я считаю, что метод argmax() будет работать для многомерных массивов, а также. Связанная документация дает пример этого:

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

Update 2

(Благодаря KennyTM «s комментарий) Вы можете использовать unravel_index(a.argmax(), a.shape), чтобы получить индекс в виде кортежа:

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

Answer 2

(edit) Я имел в виду старый ответ, который был удален. И принятый ответ пришел после моего. Я согласен с тем, что argmax лучше моего ответа.

Не было бы более читабельным/интуитивно понятным делать это?

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

Или

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