NumPy参考 >例行程序 >Indexing routines > numpy.putmask
numpy.
putmask
(a,mask,值)¶根据条件和输入值更改数组的元素。
为每个n 设置。a.flat[n] = values[n]
mask.flat[n]==True
如果值的大小与a和mask的大小不同,则它将重复。这使行为不同于。a[mask] = values
目标数组。
布尔掩码数组。它必须是相同的形状作为一个。
值投入一个地方面具是真。如果值小于a,将重复执行。
例子
>>> x = np.arange(6).reshape(2, 3)
>>> np.putmask(x, x>2, x**2)
>>> x
array([[ 0, 1, 2],
[ 9, 16, 25]])
如果值小于a,则重复:
>>> x = np.arange(5)
>>> np.putmask(x, x>1, [-33, -44])
>>> x
array([ 0, 1, -33, -44, -33])