numpy.nanargmax

numpy.nanargmax(a, axis=None)

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

参数：

a：array_like

输入数据。

axis：int，可选

要操作的轴。默认使用平展输入。

返回：

index_array：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])