numpy.argwhere()

numpy.argwhere(a) [source]

Find the indices of array elements that are non-zero, grouped by element.

Parameters:

Parameters:	a : array_like Input data.
Returns:	index_array : ndarray Indices of elements that are non-zero. Indices are grouped by element.

a : array_like

Input data.

Returns:

index_array : ndarray

Indices of elements that are non-zero. Indices are grouped by element.

Notes

np.argwhere(a) is the same as np.transpose(np.nonzero(a)).

The output of argwhere is not suitable for indexing arrays. For this purpose use where(a) instead.

Examples

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

Links:

https://docs.scipy.org/doc/numpy-1.11.0/reference/generated/numpy.argwhere.html

doc_NumPy

2025-01-10 15:47:30

Comments

Popular Articles

numpy.argsort() numpy.argsort(a, axis=-1, kind='quicksort', order=None) [source] Returns the indices that would sort an array. Perform an indirect sort along the

numpy.isscalar() numpy.isscalar(num) [source] Returns True if the type of num is a scalar type. Parameters: num : any Input argument, can be of any type and sha

numpy.mirr() numpy.mirr(values, finance_rate, reinvest_rate) [source] Modified internal rate of return. Parameters: values : array_like Cash flows (must con

numpy.flatiter class numpy.flatiter [source] Flat iterator object to iterate over arrays. A flatiter iterator is returned by x.flat for any array x. It allows i

numpy.cumsum() numpy.cumsum(a, axis=None, dtype=None, out=None) [source] Return the cumulative sum of the elements along a given axis. Parameters: a : array_li