numpy.nanargmax

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

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

Параметры

aarray_like

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

axisint, optional

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

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

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