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

在這裏，我們將檢查矩陣的每一行並計算其最小值的索引，同時忽略任何缺失值。