numpy.nanargmax

麻木的。nanargmax ( a , axis=None , out=None , * , keepdims=<无 值> )

返回指定轴中最大值的索引，忽略 NaN。对于全 NaN 切片ValueError会引发。警告：如果切片仅包含 NaN 和 -Infs，则结果不可信。

参数：

类似数组

输入数据。

轴int，可选

操作所沿的轴。默认情况下使用扁平输入。

输出数组，可选

如果提供，结果将被插入到该数组中。它应该具有适当的形状和类型。

1.22.0 版本中的新增功能。

keepdims布尔值，可选

如果将此设置为 True，则缩小的轴将作为大小为 1 的维度保留在结果中。使用此选项，结果将针对阵列正确广播。

1.22.0 版本中的新增功能。

返回：

索引数组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])