numpy.putmask

numpy.putmask(a, mask, values)

Изменяет элементы массива на основе условных значений и входных данных.

Устанавливает a.flat[n] = values[n] для каждого n, где mask.flat[n]==True.

Если values не имеет такого же размера, как a и mask, то оно будет повторяться. Это приводит к поведению, отличающемуся от a[mask] = values.

Параметры:

andarray

Массив-цель.

maskarray_like

Булевый массив-маска. Он должен иметь такую же форму, как a.

valuesarray_like

Значения, которые нужно поместить в a там, где mask равно True. Если values меньше, чем a, оно будет повторяться.

См. также

place, put, take, copyto

Примеры

>>> x = np.arange(6).reshape(2, 3)
>>> np.putmask(x, x>2, x**2)
>>> x
array([[ 0,  1,  2],
       [ 9, 16, 25]])

Если values меньше, чем a, оно повторяется:

>>> x = np.arange(5)
>>> np.putmask(x, x>1, [-33, -44])
>>> x
array([  0,   1, -33, -44, -33])