, mask, vals)[source]#

Change elements of an array based on conditional and input values.

This function uses the first N elements of vals, where N is the number of true values in mask.

  • arr (cupy.ndarray) – Array to put data into.

  • mask (array-like) – Boolean mask array. Must have the same size as a.

  • vals (array-like) – Values to put into a. Only the first N elements are used, where N is the number of True values in mask. If vals is smaller than N, it will be repeated, and if elements of a are to be masked, this sequence must be non-empty.


>>> arr = np.arange(6).reshape(2, 3)
>>>, arr>2, [44, 55])
>>> arr
array([[ 0,  1,  2],
       [44, 55, 44]])


This function may synchronize the device.

See also