numpy.place

numpy.place(arr, mask, vals)

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

类似于 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]])