如何同时遍历两个栅格目录并在循环外访问文件且保留目录结构？

我懂你现在的困扰——要同时遍历存初始图像和掩码的两个目录，既要做相关计算，又想在循环外面也能访问这些文件，还得保住原有的目录结构，嵌套循环又觉得没必要，之前试了在循环外建空堆栈往里填也没成功，对吧？

下面给你一套实用的解决方案，核心是先把两个目录的文件按路径对应好，之后不管是循环内处理还是循环外调用都很方便：

核心思路是先把两个目录下的文件按「相对路径」一一匹配，这样既能保留原目录结构，又能在循环外轻松访问任意一组图像和掩码，完全不用嵌套循环。

第一步：收集两个目录的文件映射 #

先写个小工具函数，遍历目录把每个文件的「相对路径（相对于根目录）」和「绝对路径」存成字典，这样就能用相对路径当“钥匙”，匹配另一个目录里的对应文件。用Python的话可以这么写：

import os

def build_file_map(root_dir):
    file_map = {}
    # 遍历目录下所有文件
    for root, _, files in os.walk(root_dir):
        for file in files:
            # 计算相对路径，用来匹配另一个目录的对应文件
            relative_path = os.path.relpath(os.path.join(root, file), root_dir)
            # 把相对路径作为key，绝对路径作为value存起来
            file_map[relative_path] = os.path.join(root, file)
    return file_map

# 替换成你的实际目录路径
initial_img_dir = "/path/to/your/initial_images"
mask_dir = "/path/to/your/masks"

# 获取两个目录的文件映射
initial_img_map = build_file_map(initial_img_dir)
mask_map = build_file_map(mask_dir)

第二步：配对图像和掩码 #

现在两个字典的key都是相对路径，只要key相同，就是一组对应的初始图像和掩码。我们把它们整理成一个列表（或者字典），这样循环内外都能直接用：

# 整理成包含路径和相对结构的配对列表
paired_files = []
for rel_path in initial_img_map:
    if rel_path in mask_map:
        paired_files.append({
            "initial_img_path": initial_img_map[rel_path],
            "mask_path": mask_map[rel_path],
            "relative_path": rel_path  # 保留原目录结构的相对路径
        })
    else:
        print(f"提示：找不到对应掩码的图像 -> {rel_path}")

第三步：循环内处理 + 循环外访问 #

现在paired_files就是所有对应好的文件集合，不管是循环里计算还是循环外调用都很灵活：

循环内执行计算 #

# 假设你有加载栅格图像的函数load_raster
def load_raster(file_path):
    # 替换成你实际的栅格加载逻辑，比如用rasterio/gdal
    pass

# 循环处理每一组图像和掩码
for file_pair in paired_files:
    img = load_raster(file_pair["initial_img_path"])
    mask = load_raster(file_pair["mask_path"])
    # 在这里执行你的计算逻辑...

循环外访问任意文件 #

如果之后需要在循环外调用某组特定的图像和掩码，直接通过相对路径找就行：

# 比如要找原目录结构里subdir下的image1.tif
target_rel_path = "subdir/image1.tif"
# 从配对列表里筛选出目标
target_pair = next(pair for pair in paired_files if pair["relative_path"] == target_rel_path)
# 加载使用
target_img = load_raster(target_pair["initial_img_path"])
target_mask = load_raster(target_pair["mask_path"])

为什么你之前的空堆栈方法没成功？ #

大概率是因为栅格数据是内存中的数组，你在循环内加载后直接赋值给外部的空堆栈，只会覆盖之前的值，而不是把所有栅格都存进去。如果想提前把所有栅格加载到内存（文件数量不多的情况），可以在配对的时候直接把数据存进去：

# 提前加载所有栅格数据到配对列表
for file_pair in paired_files:
    file_pair["initial_img_data"] = load_raster(file_pair["initial_img_path"])
    file_pair["mask_data"] = load_raster(file_pair["mask_path"])

# 循环外直接访问数据
print(paired_files[0]["initial_img_data"].shape)

这样既保留了原目录结构，又能在循环内外自由访问，完全不用嵌套循环~

内容的提问来源于stack exchange，提问作者Aldi Kasse 2