numpy.nanargmax#

numpy.nanargmax(a, axis=None, out=None, *, keepdims= значение>)[источник]#

Возвращает индексы максимальных значений по указанной оси, игнорируя NaN. Для срезов, полностью состоящих из NaN, ValueError возникает. Предупреждение: результатам нельзя доверять, если срез содержит только NaN и -Inf.

Параметры:

aarray_like: Входные данные.
осьint, необязательный: Ось, вдоль которой выполняется операция. По умолчанию используется сглаженный вход.
выходмассив, опционально: Если предоставлен, результат будет вставлен в этот массив. Он должен иметь соответствующую форму и dtype.

Новое в версии 1.22.0.
keepdimsbool, необязательно: Если установлено значение True, оси, которые были сокращены, остаются в результате как измерения с размером один. С этой опцией результат будет корректно транслироваться относительно массива.

Новое в версии 1.22.0.

Возвращает:

index_arrayndarray: Массив индексов или одно значение индекса.

Смотрите также

argmax, nanargmin

Примеры

>>> import numpy as np
>>> a = np.array([[np.nan, 4], [2, 3]])
>>> np.argmax(a)
0
>>> np.nanargmax(a)
1
>>> np.nanargmax(a, axis=0)
array([1, 0])
>>> np.nanargmax(a, axis=1)
array([1, 1])