如何编写While循环实现网格节点邻域Z值判定与路径节点掩模

核心逻辑调整 #

你要实现的功能属于带阈值过滤的连通域遍历，更适合用广度优先搜索（BFS）实现，比单节点计数器的while循环逻辑更清晰，也能避免漏处理多分支路径和死循环问题。
你现有的neighbors函数存在两个明显问题：

1. 没有做边界校验，当x/y处于网格边缘时，负数索引会导致取值异常
2. 仅返回邻域的数值切片，无法对应回原数组的坐标位置，没法做掩模标记和后续遍历

完整可运行代码 #

假设你的输入数组为numpy格式，结构为(X轴长度, Y轴长度, 1)，每个坐标点的Z值存在第三维度的第一个位置，可根据你的实际数组结构调整Z值的取值逻辑：

import numpy as np
from collections import deque

def get_valid_neighbors(x, y, max_x, max_y):
    """返回(x,y)的8邻域中，处于网格边界内的坐标列表"""
    neighbors = []
    for dx in (-1, 0, 1):
        for dy in (-1, 0, 1):
            # 跳过自身节点
            if dx == 0 and dy == 0:
                continue
            nx = x + dx
            ny = y + dy
            # 校验坐标在合法范围内
            if 0 <= nx < max_x and 0 <= ny < max_y:
                neighbors.append((nx, ny))
    return neighbors

def threshold_path_mask(arr, start_xy, end_xy, z_min, z_max):
    """
    输出符合条件的掩模数组
    参数:
        arr: 输入三维数组，shape为(max_x, max_y, z值维度)
        start_xy: 起始点坐标，元组格式(x,y)
        end_xy: 终点坐标，元组格式(x,y)
        z_min: Z值阈值下限
        z_max: Z值阈值上限
    返回:
        mask: 布尔型掩模，符合条件的节点为True
    """
    max_x, max_y = arr.shape[0], arr.shape[1]
    # 初始化掩模和访问标记数组
    mask = np.zeros((max_x, max_y), dtype=bool)
    visited = np.zeros((max_x, max_y), dtype=bool)
    # BFS队列，初始加入起点
    q = deque()
    
    # 先校验起点本身是否符合条件
    sx, sy = start_xy
    if z_min < arr[sx, sy, 0] < z_max:
        mask[sx, sy] = True
        visited[sx, sy] = True
        q.append(start_xy)
    else:
        # 起点本身不符合条件直接返回空掩模
        return mask
    
    while q:
        current_x, current_y = q.popleft()
        # 到达终点直接终止遍历
        if (current_x, current_y) == end_xy:
            break
        # 遍历所有合法邻域
        for nx, ny in get_valid_neighbors(current_x, current_y, max_x, max_y):
            # 未访问过且Z值符合条件
            if not visited[nx, ny] and (z_min < arr[nx, ny, 0] < z_max):
                mask[nx, ny] = True
                visited[nx, ny] = True
                q.append((nx, ny))
    
    return mask

使用示例 #

# 生成测试数组：10x10网格，Z值为x+y
test_arr = np.zeros((10,10,1))
for x in range(10):
    for y in range(10):
        test_arr[x,y,0] = x + y

# 起点(0,0)，终点(7,7)，Z值阈值为3到12
mask = threshold_path_mask(test_arr, (0,0), (7,7), 3, 12)
# 查看掩模结果
print(mask.astype(int))

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