numpy.nanargmin

numpy.nanargmin(a, axis=None, out=None, *, keepdims=<no value>)

返回指定轴中忽略 NaNs 的最小值的索引.对于全为 NaN 的切片,会引发 ValueError.警告:如果一个切片仅包含 NaNs 和 Infs,则结果不可信.

参数:

aarray_like

输入数据.

axisint, 可选

操作沿着的轴.默认使用展平的输入.

out数组, 可选

如果提供,结果将被插入到这个数组中.它应该具有适当的形状和数据类型.

在 1.22.0 版本加入.

keepdimsbool, 可选

如果设置为 True,被减少的轴将作为尺寸为1的维度保留在结果中.使用此选项,结果将正确地与数组广播.

在 1.22.0 版本加入.

返回:

index_arrayndarray

一个索引数组或单个索引值.

参见

argmin, nanargmax

示例

>>> import numpy as np
>>> 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])