numpy.ndarray.argmax

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

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

Обратитесь к numpy.argmax для полной документации.

См. также

numpy.argmax

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