使用os.walk()遍历目录树时判断当前是否为指定目录并收集文件路径

问题原因 #

你现有代码的核心问题是对os.walk的返回值含义理解错误：
os.walk每次迭代返回的三个参数分别是 当前遍历到的目录路径、当前目录下的子目录名列表、当前目录下的文件名列表，你代码中用第二个返回值（子目录列表）去和字符串"root/dir2/1"对比，逻辑完全不成立。

修正后代码 #

import os

# 统一转绝对路径避免相对路径的匹配歧义
TARGET_DIR = os.path.abspath("../../data/root/dir2/1")
SCAN_ROOT = "../../data/"
paths = []

for curr_path, _, filenames in os.walk(SCAN_ROOT):
    if os.path.abspath(curr_path) == TARGET_DIR:
        for filename in filenames:
            paths.append(os.path.join(curr_path, filename))
        # 匹配到目标目录后可直接终止遍历
        break

补充注意事项 #

• 不要用dir作为变量名，这是Python内置函数名，会覆盖原有功能，不需要用到的返回值建议用下划线_代替。
• 如果需要适配跨操作系统场景，不要写死路径分隔符/，统一用os.path.join拼接路径片段。
• 如果不想用绝对路径匹配，也可以判断路径后缀：if curr_path.endswith(os.path.join("root", "dir2", "1"))。

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