如何在NumPy数组中筛选无“非零夹0”空洞的行

核心判定规则 #

按照需求，只要单行内出现任意一组连续的「非零值-0-非零值」三元组，就判定该行存在空洞，需要排除。边界位置（行首第一列、行尾最后一列）的0不可能被两侧非零值夹持，无需纳入判定范围。

向量化实现逻辑 #

不需要逐行写循环遍历，直接用NumPy的布尔掩码完成批量判定，运行效率远高于逐元素遍历的循环写法：

• 先生成全量非零值的布尔掩码，标记数组中哪些位置是非零元素
• 截取数组的中间区域（去掉首列和尾列），同时对比每个中间位置的左侧、右侧相邻元素的掩码状态
• 满足「当前位置是0 + 左侧为非零 + 右侧为非零」的位置，就是要找的空洞点
• 沿行方向做聚合：只要某行存在任意一个空洞点，就把这行标记为需要排除的行，最终取反掩码筛选保留行

可直接运行的代码 #

import numpy as np

M = np.array([[0,10,0,20,30],[15,0,0,25,35],[0,40,40,40,0],[50,0,50,0,50]])

# 生成非零元素掩码
nonzero_mask = M != 0
# 判定每行是否存在空洞：三个条件同时满足即命中空洞
has_hole = (
    ~nonzero_mask[:, 1:-1]  # 当前位置取值为0
    & nonzero_mask[:, :-2]   # 左侧相邻位置为非零
    & nonzero_mask[:, 2:]    # 右侧相邻位置为非零
).any(axis=1)
# 筛选无空洞的行
filtered_M = M[~has_hole]
print(filtered_M)

运行输出验证 #

代码运行后得到的结果和需求完全一致：

[[15  0  0 25 35]
 [ 0 40 40 40  0]]

特殊场景适配说明 #

• 连续多个0的行（比如[1,0,0,0,1]）不会被误判：因为连续0段里的所有0位置，要么左侧是0、要么右侧是0，不会形成「非零-0-非零」的结构，符合保留规则
• 行首、行尾的0不会被误判：因为没有相邻的外侧元素，不可能形成被夹持的结构，比如[0,1,1,1,0]这类全段非零夹着首尾0的行，会被正常保留

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