Python递归遍历文件夹找指定扩展名路径

本文深入解析了Python中使用os.walk()递归遍历文件夹并精准定位指定扩展名文件的核心实践，直击初学者普遍遭遇的“路径为空”痛点——问题根源并非代码逻辑错误，而是相对路径依赖当前工作目录导致的定位失效；文章系统提供了三种可靠解决方案：绝对路径（调试首选）、基于__file__动态构建路径（部署推荐）、以及谨慎使用的os.chdir临时切换，并同步给出pathlib.rglob的现代简洁替代方案，辅以可直接运行的健壮示例和关键避坑提示（如大小写敏感、空格路径、符号链接处理等），助你彻底摆脱“找不到文件”的困扰，实现跨环境稳定、可复用的文件路径检索。

本文详解如何用 os.walk() 正确扫描目录树、匹配文件扩展名并返回绝对路径，重点解决路径错误、工作目录不匹配导致“找不到文件”的常见问题，并提供可直接运行的健壮示例代码。

本文详解如何用 os.walk() 正确扫描目录树、匹配文件扩展名并返回绝对路径，重点解决路径错误、工作目录不匹配导致“找不到文件”的常见问题，并提供可直接运行的健壮示例代码。

在 Python 中，os.walk() 是遍历目录树最标准、最可靠的方式。但初学者常遇到“函数执行无返回”或“列表为空”的问题——这几乎从不源于代码逻辑错误，而是路径配置不当或当前工作目录（Current Working Directory）与目标路径不一致所致。

你的原始代码逻辑完全正确：

import os

def retrieve_files(file_extension):
    file_paths = []
    for root, dirs, files in os.walk("Assignment8"):  # ← 关键：此处路径是相对路径
        for file in files:
            if file.endswith(file_extension):
                file_paths.append(os.path.join(root, file))
    return file_paths

问题核心在于 "Assignment8" 是一个相对路径。Python 会以你运行脚本时所在的终端/命令行目录为基准去查找该文件夹。例如：

• 若你在 ~/Desktop/ 下执行 python main.py，Python 就会在 ~/Desktop/Assignment8/ 中搜索；
• 但你的实际结构是 ~/Desktop/python 2/FOLDER/contents/baby_names/...，且 Assignment8 并不存在于 Desktop 下 —— 因此返回空列表。

✅ 正确做法有三种，推荐按优先级使用：

1. 使用绝对路径（最稳妥，适合开发调试）

明确指定磁盘根路径，避免歧义。注意 Windows 下使用双反斜杠 \\ 或原始字符串 r""：

import os

def retrieve_files(file_extension):
    # ✅ 推荐：使用绝对路径（替换为你的真实路径）
    target_dir = r"N:\Code"  # 或 "/home/user/Projects/Assignment8"
    file_paths = []
    for root, dirs, files in os.walk(target_dir):
        for file in files:
            if file.endswith(file_extension):
                file_paths.append(os.path.join(root, file))
    return file_paths

# 示例：查找所有 .txt 文件（如 yob1880.txt, yob1881.txt）
txt_files = retrieve_files(".txt")
for path in txt_files:
    print(path)

2. 基于 __file__ 动态计算相对路径（推荐用于项目部署）

让脚本自动定位自身所在目录，再向上/向下拼接目标路径，彻底摆脱手动路径依赖：

import os

def retrieve_files(file_extension, base_folder="baby_names"):
    # 获取 main.py 所在目录（即 FOLDER/）
    current_dir = os.path.dirname(os.path.abspath(__file__))
    # 拼接为 FOLDER/contents/baby_names/
    target_dir = os.path.join(current_dir, "contents", base_folder)

    file_paths = []
    for root, dirs, files in os.walk(target_dir):
        for file in files:
            if file.endswith(file_extension):
                file_paths.append(os.path.join(root, file))
    return file_paths

# 调用示例（无需关心当前工作目录在哪）
txt_files = retrieve_files(".txt", base_folder="baby_names")
print(f"找到 {len(txt_files)} 个 .txt 文件：")
for p in txt_files:
    print(f"  → {p}")

3. 显式切换工作目录（谨慎使用）

仅作临时调试，不建议写入生产代码：

import os
os.chdir(r"/full/path/to/your/FOLDER")  # 切换到目标父目录
# 然后再调用原版 retrieve_files("Assignment8")

⚠️ 重要注意事项

• 不要硬编码含空格或特殊字符的路径（如 "python 2"），易引发解析错误；建议重命名文件夹为 python2 或使用引号包裹。
• file.endswith(".txt") 区分大小写；若需兼容 .TXT，改用 file.lower().endswith(".txt")。
• os.walk() 默认不进入软链接目录；如需遍历符号链接，传入 followlinks=True 参数（注意循环引用风险）。
• 对于超大目录，可考虑 pathlib.Path.rglob()（Python 3.5+）实现更简洁、面向对象的写法：

from pathlib import Path

def retrieve_files_pathlib(file_extension, base_folder="baby_names"):
    current_dir = Path(__file__).parent
    target_dir = current_dir / "contents" / base_folder
    return [str(p) for p in target_dir.rglob(f"*{file_extension}")]

# 一行等效：list(map(str, Path("N:/Code").rglob("*.txt")))

总结：os.walk() 本身无缺陷，99% 的“查不到文件”问题都源于路径理解偏差。始终优先使用绝对路径验证功能，再迁移到基于 __file__ 的动态路径方案，即可一劳永逸地解决跨环境路径适配难题。

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于文章的相关知识，也可关注golang学习网公众号。