dpnp.flatnonzero

dpnp.flatnonzero(a)

Return indices that are non-zero in the flattened version of a.

This is equivalent to dpnp.nonzero(dpnp.ravel(a))[0].

For full documentation refer to numpy.flatnonzero.

Parameters:

a ({dpnp.ndarray, usm_ndarray}) -- Input data.

Returns:

out -- Output array, containing the indices of the elements of a.ravel() that are non-zero.

Return type:

dpnp.ndarray

See also

dpnp.nonzero

Return the indices of the non-zero elements of the input array.

dpnp.ravel

Return a 1-D array containing the elements of the input array.

Examples

>>> import dpnp as np
>>> x = np.arange(-2, 3)
>>> x
array([-2, -1,  0,  1,  2])
>>> np.flatnonzero(x)
array([0, 1, 3, 4])

Use the indices of the non-zero elements as an index array to extract these elements:

>>> x.ravel()[np.flatnonzero(x)]
array([-2, -1,  1,  2])