遍历目录并按特定规则向字典追加内容的技术实现求助
我来帮你搞定这个批量处理文件的需求!针对你这种要遍历几百个子目录、提取特定文件夹下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
相关产品推荐
相关产品推荐

