Python os.walk详解|极客笔记

Python os.walk详解

在Python中，os.walk是一个非常方便的方法，用于遍历目录树，返回该目录及其子目录中的文件和文件夹。本文将详细解释os.walk的用法，并提供一些实用的示例。

`os.walk`方法概述

os.walk方法是Python中的一个递归遍历函数，用于遍历指定目录下的所有子目录和文件。它返回一个生成器对象，其中包含当前目录路径、当前目录下所有子目录的名称以及当前目录下所有文件的名称。

os.walk方法的语法如下：

for root, dirs, files in os.walk(top, topdown=True, onerror=None, followlinks=False):
    # 迭代操作

参数说明：

top：需要遍历的根目录路径。
topdown：可选参数，默认值为True，表示遍历目录时，先遍历父目录再遍历子目录；若设为False，则会先遍历子目录再遍历父目录。
onerror：可选参数，用于处理遍历过程中的错误。
followlinks：可选参数，默认值为False，如果设为True，则会跟踪软链接所指的目录。

示例代码

接下来我们通过一些示例代码来更好地理解os.walk方法的使用。

import os

# 遍历当前目录下的所有文件和子目录
for root, dirs, files in os.walk(".", topdown=True):
    print("当前目录路径：", root)
    print("子目录列表：", dirs)
    print("文件列表：", files)
    print()

运行结果：

当前目录路径： .
子目录列表： ['dir1', 'dir2']
文件列表： ['file1.txt', 'file2.txt']

当前目录路径： .\dir1
子目录列表： ['subdir1']
文件列表： ['file3.txt']

当前目录路径： .\dir1\subdir1
子目录列表： []
文件列表： ['file4.txt']

当前目录路径： .\dir2
子目录列表： []
文件列表： ['file5.txt']

在上面的示例中，我们遍历了当前目录下的所有文件和子目录，并分别打印出了当前目录路径、子目录列表和文件列表。可以看到os.walk返回了目录结构的相关信息。

示例2：搜索指定后缀名的文件

有时候我们需要搜索特定后缀名的文件，下面是一个示例：

import os

suffix = '.txt'

for root, dirs, files in os.walk(".", topdown=True):
    for file in files:
        if file.endswith(suffix):
            print(os.path.join(root, file))

这段代码会遍历当前目录及其子目录中所有的.txt文件，并打印出它们的完整路径。