numpy.ndarray.argmax

метод

ndarray.argmax(axis=None, out=None, *, keepdims=False)

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

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

См. также

numpy.argmax

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