numpy.nanargmax#
- numpy.nanargmax(a, axis=None, out=None, *, keepdims=<no value>)[source]#
返回指定轴上最大值的索引,忽略 NaN。对于全 NaN 切片,会引发
ValueError
。警告:如果切片仅包含 NaN 和 -Inf,则无法信任结果。- 参数:
- aarray_like
输入数据。
- axisint, 可选
操作的轴。默认情况下使用扁平化的输入。
- outarray, 可选
如果提供,结果将插入到此数组中。它应该具有合适的形状和数据类型。
版本 1.22.0 中的新功能。
- keepdimsbool, 可选
如果将其设置为 True,则减少的轴将保留在结果中,作为大小为一的维度。使用此选项,结果将针对数组正确广播。
版本 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])