numpy - 如何理解 np.argwhere 函数？

Question

Signature: np.argwhere(a)
Docstring:
Find the indices of array elements that are non-zero, grouped by element.

例子

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

“非零”和“按元素分组”是什么意思？什么是“x>1”？

Answer 1

在每一行中，第一个条目是行索引，第二个条目是满足条件的 x 条目的列索引。

例如:
2 大于 1
所以 argwhere 的第一行给你 [0, 2]指向 x 中 2 的位置。