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

例子

基本用法

x = np.array([1,2,3])
np.argwhere(x)



array([[0],
       [1],
       [2]])

此處，由於 x 中的所有值均非零，因此返回所有索引。請注意，返回的數組是二維的。

使用麵罩

在 NumPy 中，False 值相當於 0，而 True 值相當於 1。這意味著 argwhere(~) 可用於掩碼。

假設我們想要確定大於 2 的值的索引。我們可以這樣做：

x = np.array([1,2,3,4])
mask = x > 2



array([False, False,  True,  True])

使用麵膜可以得到：

np.argwhere(mask)



array([[2],
       [3]])

二維數組

假設我們想要確定 2D Numpy 數組中大於 2 的值的索引

x = np.array([[1,2],[3,4],[5,6]])
mask = x > 2



array([[False, False],
       [ True,  True],
       [ True,  True]])

使用麵膜可以得到：

np.argwhere(mask)



array([[1, 0],
       [1, 1],
       [2, 0],
       [2, 1]])

對此的解釋方法如下：

[1,0] -> row 1, column 0 is non-zero (i.e. greater than 2)
[1,1] -> row 1, column 1 is non-zero
[2,0] -> row 2, column 0 is non-zero
[2,1] -> row 2, column 1 is non-zero