numpy.nanargmax#
- numpy.nanargmax(a, axis=None, out=None, *, keepdims=<no value>)[源代码]#
返回指定轴上忽略 NaN 的最大值的索引. 对于全 NaN 切片,会引发
ValueError. 警告:如果切片仅包含 NaN 和 -Infs,则结果不可信.- 参数:
- aarray_like
输入数据.
- 轴int, optional
要沿其操作的轴. 默认情况下,使用扁平化输入.
- 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])