numpy.nanargmax
numpy.nanargmax(a, axis=None) [source]
返回指定轴上最大值的索引,忽略NaN。 对于所有NaN切片,会引发ValueError
。 警告:如果切片仅包含NaN和-Infs,则结果不可信任。
参数 : | a :array_like 输入数据。 axis : 沿其运行的轴。 默认情况下,使用拼合输入。 |
返回值 : | index_array : 索引数组或单个索引值。 |
例子
>>> 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])