Python NumPy nanargmax方法用法及代碼示例

例子

一維數組

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



1

這裏，返回 1，因為最大值(即 5)位於索引 1。相反，np.argmax(x) 方法將返回 0，因為它認為 NaN 是最大的。

二維數組

假設我們有以下 2D Numpy 數組：

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



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

整個數組的最大索引

要獲取整個數組中最大值的索引，請省略 axis 參數：

np.nanargmax(x)



1

每列的最大索引

要獲取按列的最大值的索引，請設置 axis=0 ：

np.nanargmax(x, axis=0)



array([1, 0])

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

每行的最大索引

要按行獲取最大值的索引，請設置 axis=1 ：

np.nanargmax(x, axis=1)



array([1, 1])

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