Python numpy argwhere用法及代碼示例

本文簡要介紹 python 語言中 numpy.argwhere 的用法。

用法:

numpy.argwhere(a)

查找按元素分組的非零數組元素的索引。

參數：

a： array_like

輸入數據。

返回：

index_array： (N, a.ndim) ndarray

非零元素的索引。索引按元素分組。該數組將具有 (N, a.ndim) 形狀，其中 N 是非零項目的數量。

注意：

np.argwhere(a) 與 np.transpose(np.nonzero(a)) 幾乎相同，但會為 0D 數組生成正確形狀的結果。

argwhere 的輸出不適合索引數組。為此，請改用nonzero(a)。

例子：

>>> x = np.arange(6).reshape(2,3)
>>> x
array([[0, 1, 2],
       [3, 4, 5]])
>>> np.argwhere(x>1)
array([[0, 2],
       [1, 0],
       [1, 1],
       [1, 2]])