上一主题

numpy.argmin

下一主题

numpy.argwhere

numpy.nanargmin

numpy.nanargmin(a, axis=None)[source]

返回指定轴中忽略NaN的最小值的索引。对于所有NaN片,ValueError被引发。警告:如果切片只包含NaN和Inf,则结果不可信任。

参数:

a:array_like

输入数据。

axis:int,可选

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

返回:

index_array:ndarray

数组索引或单个索引值。

也可以看看

argminnanargmax

例子

>>> a = np.array([[np.nan, 4], [2, 3]])
>>> np.argmin(a)
0
>>> np.nanargmin(a)
2
>>> np.nanargmin(a, axis=0)
array([1, 1])
>>> np.nanargmin(a, axis=1)
array([1, 0])