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