numpy.argmax

numpy.argmax(a, axis=None, out=None) [source]

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

Параметры:	a : array_like Входной массив. axis : int, optional По умолчанию индекс в сплющенном массиве, в противном случае вдоль указанной оси. out : array, optional Если указано, результат будет вставлен в этот массив. Он должен иметь соответствующую форму и тип.
Возвращает:	index_array : ndarray of ints Массив индексов в массиве. Он имеет ту же форму, что и a.shape, с извлечённым измерением вдоль axis.

См. также

ndarray.argmax, argmin

amax

Максимальное значение вдоль заданной оси.

unravel_index

Преобразование плоского индекса в кортеж индексов.

Примечания

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

Примеры

>>> a = np.arange(6).reshape(2,3) + 10
>>> a
array([[10, 11, 12],
       [13, 14, 15]])
>>> np.argmax(a)
5
>>> np.argmax(a, axis=0)
array([1, 1, 1])
>>> np.argmax(a, axis=1)
array([2, 2])

Индексы максимальных элементов многомерного массива:

>>> ind = np.unravel_index(np.argmax(a, axis=None), a.shape)
>>> ind
(1, 2)
>>> a[ind]
15

>>> b = np.arange(6)
>>> b[1] = 5
>>> b
array([0, 5, 2, 3, 4, 5])
>>> np.argmax(b)  # Only the first occurrence is returned.
1