dpnp.where

dpnp.where(condition, x=None, y=None, /)

Return elements chosen from x or y depending on condition.

When only condition is provided, this function is a shorthand for dpnp.nonzero(condition).

For full documentation refer to numpy.where.

Returns:

y – An array with elements from x where condition is True, and elements from y elsewhere.

Return type:

dpnp.ndarray

Limitations

Parameter condition is supported as either dpnp.ndarray or dpctl.tensor.usm_ndarray. Parameters x and y are supported as either scalar, dpnp.ndarray or dpctl.tensor.usm_ndarray Otherwise the function will be executed sequentially on CPU. Input array data types of x and y are limited by supported DPNP Data types.

See also

nonzero

The function that is called when x and `y`are omitted.

Examples

>>> import dpnp as dp
>>> a = dp.arange(10)
>>> d
array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
>>> dp.where(a < 5, a, 10*a)
array([ 0,  1,  2,  3,  4, 50, 60, 70, 80, 90])