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: numpy.ndarray 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]])