如何根据布尔掩码提取NumPy数组元素并将掩码为False的位置置0？

针对你的需求，这里有三种实用的实现方式：

• 方法1：复制数组后通过布尔索引赋值
  先复制原数组避免修改原始数据，再对掩码取反定位需要设0的位置：

  import numpy as np
  
  a = np.array([1,2,3,4,5])
  mask = np.array([True, True, True, False, False])
  
  result = a.copy()
  result[~mask] = 0  # ~mask 对掩码取反，定位False的位置
  

• 方法2：利用布尔值与数组的乘法特性
  NumPy会自动将布尔值转换为整数（True→1，False→0），直接相乘即可快速得到结果：

  result = a * mask
  

• 方法3：使用np.where函数
  通过条件判断直接选择对应位置的值，逻辑直观清晰：

  result = np.where(mask, a, 0)  # 满足mask为True时取a的元素，否则取0
  

以上三种方法都能得到预期结果 array([1, 2, 3, 0, 0])，可以根据实际场景选择使用。

内容的提问来源于stack exchange，提问作者pd441