os模块 | os.walk()用法

os.walk() 是 Python 标准库 os 模块中的一个函数，用于递归遍历目录结构，返回当前目录及其子目录中的所有文件和文件夹。

下面是 os.walk() 函数的用法示例：

1. 导入模块：

import os

2. 使用 os.walk() 遍历目录：

start_path = '/path/to/directory'

for root, dirs, files in os.walk(start_path):
    print("Current directory:", root)
    print("Subdirectories:", dirs)
    print("Files:", files)

在这个示例中，os.walk() 返回一个生成器，每次迭代都会产生一个三元组 (root, dirs, files)，其中：

• root 是当前目录的路径。
• dirs 是当前目录中的子目录列表。
• files 是当前目录中的文件列表。

这个循环将遍历指定目录（包括其子目录）中的所有内容，并输出每个目录的路径、子目录列表和文件列表。

3. 针对特定条件过滤文件：

for root, dirs, files in os.walk(start_path):
    for file in files:
        if file.endswith('.txt'):
            print(os.path.join(root, file))

这个示例中，我们遍历指定目录及其子目录，找出所有扩展名为 .txt 的文件并输出它们的完整路径。

os.walk() 可以在目录遍历、文件搜索、数据处理等方面非常有用。在使用时，替换 /path/to/directory 为要遍历的实际目录路径。

需要注意的是，os.walk() 是递归遍历，所以对于大型目录结构，可能会占用较多内存和时间。