Python NumPy putmask方法用法及代码示例

Numpy 的 putmask(~) 在给定布尔掩码的输入数组中设置特定值。这是就地完成的，即不会创建新的 Numpy 数组。

1. a | array-like

输入数组。所有数组都被视为展平数组。

2. mask | array-like 共 booleans

指示是否设置该值的布尔值数组。形状必须与 a 相同。

3. values | array-like

要设置的值。如果 v 比 a 短，则将重复 v 以确保形状匹配。

None - 设置就地完成。

a = np.array([5,6,7,8])
np.putmask(a, [True, False, True, False], 9)
a



array([9, 6, 9, 8])

请注意如何仅设置掩码中具有相应 True 值的值。

考虑以下二维数组：

a = np.array([[5,6],[7,8]])
a



array([[5, 6],
       [7, 8]])

np.putmask(a, [True, False, True, False], 9)
a



array([[9, 6],
       [9, 8]])

所有数组都被视为展平数组。

a = np.array([[5,6],[7,8]])
np.putmask(a, [True, True, False, False], [9,10])
a



array([[ 9, 10],
       [ 7,  8]])

相关用法

注：本文由纯净天空筛选整理自Isshin Inada大神的英文原创作品 NumPy | putmask method。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。