numpy.nanargmax

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

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

Параметры:	a : array_like Входные данные. axis : int, необязательно Ось, по которой нужно выполнить операцию. По умолчанию используется сплющенная входная информация.
Возвращает:	index_array : ndarray Массив индексов или одно значение индекса.

См. также

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