Python numpy nanargmax用法及代碼示例

本文簡要介紹 python 語言中 numpy.nanargmax 的用法。

用法:

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

返回指定軸中最大值的索引，忽略 NaN。對於all-NaN 切片，ValueError 被提升。警告：如果切片僅包含 NaNs 和 -Infs，則結果不可信。

參數：

a： array_like

輸入數據。

axis： 整數，可選

操作的軸。默認情況下使用扁平化輸入。

out： 數組，可選

如果提供，結果將被插入到這個數組中。它應該具有適當的形狀和數據類型。

keepdims： 布爾型，可選

如果將其設置為 True，則縮小的軸將作為尺寸為 1 的尺寸留在結果中。使用此選項，結果將針對數組正確廣播。

返回：

index_array： ndarray

索引數組或單個索引值。

例子：

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