numpy.putmask

numpy.putmask(a, mask, values)

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

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

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

Параметры:	a : array_like Целевой массив. mask : array_like Массив булевых масок. Он должен иметь ту же форму, что и a. values : array_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])