Python中的numpy.nanargmin

在Python的numpy模块中，numpy.nanargmin函数可以用来查找数组中的最小值的索引，而忽略NaN值。

语法

numpy.nanargmin(a, axis=None)

参数

• a：输入数组
• axis：指定数组的轴，如果不指定，则函数会查找整个数组的最小值索引，否则只沿指定轴查找

返回值

• 返回数组中最小值的索引，如果有多个，则返回最小索引值的第一个出现的位置

示例

import numpy as np

a = np.array([1, np.nan, 3, 4, 5])
print(np.nanargmin(a))  # 输出：0

在示例中，我们定义了一个数组a，其中包含NaN值。我们使用np.nanargmin()函数查找数组a中忽略NaN值的最小值的索引，并返回了索引值0。

当我们指定轴时，np.nanargmin()函数仅在指定的轴上进行比较。例如，如果我们定义一个二维数组b，则可以使用下面的代码来查找每一行中的最小值：

import numpy as np

b = np.array([[1, 2, 3], [4, 5, np.nan], [7, 8, 9]])
print(np.nanargmin(b, axis=1))  # 输出：[0 1 0]

在这个示例中，我们将轴参数指定为1，因此函数在每一行中查找最小值，并返回索引值。