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