numpy.nanargmax

numpy.nanargmax(a, axis=None, out=None, *, keepdims=<no value>)[source]

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

Параметры:

aarray_like

Входные данные.

axisint, необязательно

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

outarray, необязательно

Если указано, результат будет помещён в этот массив. Он должен иметь соответствующую форму и тип данных.

Добавлена в версии 1.22.0.

keepdimsbool, необязательно

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

Добавлена в версии 1.22.0.

Возвращаемые значения:

index_arrayndarray

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

См. также

argmax, nanargmin

Примеры

>>> 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])