argwhere¶
-
sparse.
argwhere
(a)[source]¶ Find the indices of array elements that are non-zero, grouped by element.
- Parameters
a (array_like) – Input data.
- Returns
index_array
- Return type
See also
Examples
>>> import sparse >>> x = sparse.COO(np.arange(6).reshape((2, 3))) >>> sparse.argwhere(x > 1) array([[0, 2], [1, 0], [1, 1], [1, 2]])