You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

遍历目录并按特定规则向字典追加内容的技术实现求助

我来帮你搞定这个批量处理文件的需求!针对你这种要遍历几百个子目录、提取特定文件夹下PNG路径并整理到字典的场景,用Python实现是最省心的,下面给你两种实用方案,你可以按需选择:

方案1:使用os模块(传统路径处理方式)

os模块是Python处理文件系统的基础工具,用os.walk可以递归遍历目录树,非常适配你的场景:

import os

# 替换成你的MAIN目录实际路径
main_directory = "MAIN"
# 初始化存储结果的字典
png_path_dict = {}

# 递归遍历MAIN下的所有目录
for root, dirs, files in os.walk(main_directory):
    # 判断当前目录是否是我们要找的subfolder_01
    if os.path.basename(root) == "subfolder_01":
        # 遍历当前目录下的所有文件
        for file_name in files:
            # 筛选出PNG文件
            if file_name.endswith(".png"):
                # 拼接完整文件路径
                full_path = os.path.join(root, file_name)
                # --------------------------
                # 这里根据你的特定规则生成字典的键和值
                # 示例1:用PNG文件名(不含后缀)作为键
                dict_key = os.path.splitext(file_name)[0]
                # 示例2:用上一层目录名(比如folder_01)作为键
                # dict_key = os.path.basename(os.path.dirname(root))
                # --------------------------
                png_path_dict[dict_key] = full_path

# 查看结果
print(png_path_dict)
方案2:使用pathlib模块(现代路径处理方式)

pathlib是Python3.4+引入的现代路径处理库,语法更简洁直观,用rglob可以直接递归匹配特定路径的文件:

from pathlib import Path

# 替换成你的MAIN目录实际路径
main_directory = Path("MAIN")
png_path_dict = {}

# 递归查找所有subfolder_01下的PNG文件
for png_file in main_directory.rglob("subfolder_01/*.png"):
    # --------------------------
    # 同样根据你的规则生成键值对
    # 示例1:用PNG文件名(不含后缀)作为键
    dict_key = png_file.stem
    # 示例2:用上一层目录名(比如folder_01)作为键
    # dict_key = png_file.parent.parent.name
    # --------------------------
    png_path_dict[dict_key] = str(png_file)

# 查看结果
print(png_path_dict)

一些实用提示

  • 两种方案都能高效处理650+子目录的场景,pathlib的代码更简洁易读,推荐优先使用
  • 关于你提到的「特定规则」,你只需要修改代码中生成dict_key的部分即可,比如如果需要关联folder编号和PNG文件,就用示例2的方式
  • 如果需要处理重复键的情况(比如不同folder下有同名PNG),可以把值改成列表,比如png_path_dict.setdefault(dict_key, []).append(full_path)
  • 这两种方法都是Python原生实现,处理几千个PNG文件完全没有性能问题

内容的提问来源于stack exchange,提问作者Lukasz

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.05.21 06:43:56