numpy.place

numpy.place(arr, mask, vals)

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

Similar to np.putmask(a, mask, vals), the difference is that place uses the first N elements of vals, where N is the number of True values in mask, while putmask uses the elements where mask is True.

Note that extract does the exact opposite of place.

Parameters:

a : array_like

Array to put data into.

mask : array_like

Boolean mask array. Must have the same size as a.

vals : 1-D sequence

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.

See also

putmask, put, take, extract

Examples

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