numpy.nanargmax

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

傳回指定軸中最大值的索引，忽略 NaN。對於全為 NaN 的切片，會引發 ValueError。警告：如果切片僅包含 NaN 和 -Inf，則結果不可信任。

參數:

aarray_like

輸入資料。

axisint, optional

要沿其運算的軸。預設使用扁平化輸入。

outarray, optional

如果提供，結果將插入此陣列。它應具有適當的形狀和 dtype。

版本 1.22.0 新增。

keepdimsbool, optional

如果設定為 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])