numpy.place#

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

根据条件和输入值更改数组的元素。

np.copyto(arr, vals, where=mask) 类似,不同之处在于 place 使用 vals 的前 N 个元素(N 是 mask 中 True 值的数量),而 copyto 使用 mask 为 True 的元素。

请注意,extract 的作用与 place 完全相反。

参数:
arrndarray

要放入数据的数组。

maskarray_like

布尔掩码数组。必须与 a 的大小相同。

vals1-D 序列

要放入 a 的值。只使用前 N 个元素,其中 N 是 mask 中 True 值的数量。如果 vals 小于 N,它将被重复;如果要掩码 a 的元素,则此序列不能为空。

另请参见

copyto, put, take, extract

示例

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