numpy.putmask

numpy.putmask(a, mask, values)

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

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

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

Параметры

aarray_like

Целевой массив.

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])