如何使用np.ma模块屏蔽数组中指定索引区间内的元素？

完全可以用np.ma模块实现这个需求，核心是创建对应索引区间的掩码，再将其应用到数组上。以下是具体实现步骤：

步骤1：导入依赖模块 #

import numpy as np
import numpy.ma as ma

步骤2：创建原始数组 #

以包含365个元素的数组为例：

original_array = np.arange(365)  # 生成0到364的数组，共365个元素

步骤3：构造掩码数组 #

初始化一个全为False的掩码（False表示不屏蔽，True表示需要屏蔽），然后将索引170到200的区间设为True：

mask = np.zeros(len(original_array), dtype=bool)
mask[170:201] = True  # 注意切片左闭右开，所以要写到201才能覆盖索引200

步骤4：生成屏蔽后的数组 #

使用ma.masked_array将原始数组和掩码结合：

masked_array = ma.masked_array(original_array, mask=mask)

验证结果 #

• 查看屏蔽后的数组，索引170-200的元素会被标记为屏蔽：

print(masked_array[168:202])

输出会显示类似：[168 169 -- -- ... -- 201 202]，其中--代表被屏蔽的元素。

• 提取所有未被屏蔽的元素，可使用compressed()方法：

unmasked_elements = masked_array.compressed()
print(len(unmasked_elements))  # 输出应为365 - 31 = 334（170到200共31个元素）

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