Python NumPy nanargmin方法用法及代码示例

例子

一维数组

x = np.array([5,np.NaN,1,3])
np.nanargmin(x)



2

这里，返回 2，因为最小值(即 1)位于索引 2。相反，np.argmin(x) 方法将返回 1，因为它认为 NaN 是最小的。

二维数组

假设我们有以下 2D Numpy 数组：

x = np.array([[5,np.NaN],[1,3]])
x



array([[ 5., nan],
       [ 1.,  3.]])

整个数组的最小索引

要获取整个数组中最小值的索引，请省略 axis 参数：

np.nanargmin(x)



2

每列的最小索引

要按列获取最小值的索引，请设置 axis=0 ：

np.nanargmin(x, axis=0)



array([1, 1])

在这里，我们将检查矩阵的每一列并计算其最小值的索引，同时忽略任何缺失值。

每行的最小索引

要按行获取最小值的索引，请设置 axis=1 ：

np.nanargmin(x, axis=1)



array([0, 0])

在这里，我们将检查矩阵的每一行并计算其最小值的索引，同时忽略任何缺失值。