如何利用numpy.ma仅对未掩码值赋值？掩码数组值复制方法咨询

1. 仅为未掩码值赋值 #

numpy.ma的掩码数组本身就支持自动跳过掩码元素的赋值操作，无需额外复杂逻辑：

import numpy as np

# 创建示例掩码数组
masked_arr = np.ma.array([1, 2, 3, 4], mask=[False, True, False, True])

# 直接赋值，numpy.ma会自动仅修改未掩码元素
masked_arr[:] = 0
# 结果: masked_arr -> [0, --, 0, --]

# 也可以通过掩码索引精准赋值
masked_arr[~masked_arr.mask] = 10
# 结果: masked_arr -> [10, --, 10, --]

直接对掩码数组执行赋值时，numpy.ma内部会自动过滤掉被掩码的元素，只对未掩码部分生效，完全贴合掩码数组的核心设计。

2. 将掩码数组的未掩码值复制到目标数组 #

有两种简洁高效的方法，比ma.choose更符合掩码数组的使用逻辑：

方法一：利用掩码索引直接赋值 #

import numpy as np

source_masked = np.ma.array([10, 20, 30, 40], mask=[False, True, False, True])
target_arr = np.array([1, 2, 3, 4])

# 获取未掩码位置的布尔索引
unmasked_pos = ~source_masked.mask
# 仅将源数组未掩码的值复制到目标数组对应位置
target_arr[unmasked_pos] = source_masked[unmasked_pos]

# 结果: target_arr -> [10, 2, 30, 4]

方法二：使用np.ma.where #

# 用np.ma.where根据掩码选择值：掩码为True时保留目标值，否则用源值
target_arr = np.ma.where(source_masked.mask, target_arr, source_masked)
# 结果与方法一一致

这两种方法都能实现“仅复制未掩码值，目标数组掩码对应位置保持不变”的需求，逻辑清晰且效率更高，可完全替代复杂的ma.choose实现。

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