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