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