### Python 中 `os.walk` 的使用方法 #### 方法概述 `os.walk()` 是 Python 标准库中的一个重要函数，用于遍历文件系统中的目录树。它会生成一个三元组 `(root, dirs, files)`，其中： - **root**: 当前正在遍历的目录路径。 - **dirs**: 列表形式表示当前目录下的子目录名集合。 - **files**: 列表形式表示当前目录下的文件名集合。 其基本语法格式如下[^1]: ```python os.walk(top[, topdown=True[, onerror=None[, followlinks=False]]]) ``` 参数说明： - **top**: 起始遍历的根目录路径。 - **topdown**: 如果为 True，则优先遍历父目录；如果为 False，则优先遍历子目录。 - **onerror**: 可选错误处理回调函数，当遇到访问权限等问题时调用该函数。 - **followlinks**: 是否跟随符号链接，默认不跟随。 --- #### 示例代码展示 以下是几个常见的 `os.walk` 使用场景： ##### 场景一：简单遍历指定目录及其所有子目录 下面的例子展示了如何通过 `os.walk` 遍历 `/home/dechin/projects/2021-python/` 下的所有文件和目录[^2]。 ```python import os for root, dirs, files in os.walk("/home/dechin/projects/2021-python/"): print(f"当前目录: {root}") print(f"子目录列表: {dirs}") print(f"文件列表: {files}\n") ``` ##### 场景二：排除特定目录 有时我们希望在遍历时跳过某些特定目录。可以通过修改 `dirs` 列表来实现这一功能[^3]。例如，在以下代码中，我们将排除名为 `"ClassF"` 和 `"Class4K"` 的子目录。 ```python allfiles = [] def get_raw_data(rawdir): for root, dirs, files in os.walk(rawdir): # 修改 dirs 原地移除不需要的子目录 dirs[:] = [d for d in dirs if d not in ["ClassF", "Class4K"]] for f in files: if re.search(r'yuv$', f): # 这里假设只收集扩展名为 yuv 的文件 allfiles.append(os.path.join(root, f)) allfiles.sort(key=str.lower) return allfiles # 测试函数 result_files = get_raw_data("/path/to/rawdir") print(result_files) ``` ##### 场景三：限定最大遍历深度 默认情况下，`os.walk` 不支持直接设置最大遍历深度。但是可以借助外部计数器或者递归来控制层数。以下是一个简单的例子，仅允许最多两层深的遍历。 ```python from pathlib import Path def walk_with_depth(path, maxdepth=2): path = Path(path).resolve() stack = [(str(path), 0)] while stack: current_dir, depth = stack.pop() yield (current_dir, [f.name for f in Path(current_dir).iterdir() if f.is_dir()], [f.name for f in Path(current_dir).iterdir() if f.is_file()]) if depth < maxdepth: for subdir in sorted([f.name for f in Path(current_dir).iterdir() if f.is_dir()], reverse=True): full_subdir_path = str(Path(current_dir) / subdir) stack.append((full_subdir_path, depth + 1)) # 应用自定义深度限制版本 for result_tuple in walk_with_depth("/some/path/", maxdepth=2): print(result_tuple) ``` --- ### 总结 以上分别介绍了 `os.walk` 的基础用法以及一些高级技巧，包括过滤特定目录、匹配特定文件类型以及限制遍历深度等内容。这些技术可以帮助开发者更灵活高效地操作本地文件系统资源。