numpy.nanargmax#
- numpy.nanargmax(a, axis=None, out=None, *, keepdims=<no value>)[源代码]#
返回指定轴上的最大值索引,忽略 NaN. 对于全 NaN 切片,会引发
ValueError. 警告:如果切片仅包含 NaN 和 -Inf,则结果不可信.- 参数:
- aarray_like
输入数据.
- axis整数,可选
操作所在的轴. 默认情况下,使用扁平化的输入.
- outarray,可选
如果提供,结果将被插入到此数组中.它应该是适当的形状和dtype.
在 1.22.0 版本加入.
- keepdimsbool, 可选
如果设置为True,则缩减的轴将作为大小为1的维度保留在结果中.使用此选项,结果将针对数组正确广播.
在 1.22.0 版本加入.
- 返回:
- index_arrayndarray
索引数组或单个索引值.
示例
>>> 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])