numpy.ndarray.argmax

метод

ndarray.argmax(axis=None, out=None)

Возвращает индексы максимальных значений по заданной оси.

См. numpy.argmax для полной документации.

См. также

numpy.argmax

эквивалентная функция