numpy.place

numpy.place(arr, mask, vals) [source]

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

Аналогично np.copyto(arr, vals, where=mask), отличие заключается в том, что place использует первые N элементов из vals, где N — количество истинных значений в mask, в то время как copyto использует элементы, где mask равно True.

Обратите внимание, что extract делает ровно противоположное place.

Параметры:

arr : ndarray Массив, в который будут помещены данные. mask : array_like Массив булевых масок. Размер должен совпадать с размером a. vals : 1-D sequence Значения, которые будут помещены в a. Используются только первые N элементов, где N — количество истинных значений в mask. Если vals меньше N, оно будет повторяться. Если элементы a должны быть замаскированы, эта последовательность должна быть непустой.

См. также

copyto, put, take, extract

Примеры

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