Python中如何使用os模块中的walk函数遍历目录树

在Python中，可以使用os模块中的walk函数来遍历目录树。walk函数用于递归地遍历一个目录下的所有子目录和文件。

walk函数的基本语法如下：

os.walk(top, topdown=True, onerror=None, followlinks=False)

参数说明：

- top：要遍历的目录的路径。

- topdown：可选参数，默认为True。表示是否要按照从顶层到底层的顺序遍历目录。如果设置为False，则会按照从底层到顶层的顺序遍历。

- onerror：可选参数，用于处理遍历过程中出现的错误。如果不传入此参数，则默认会抛出异常。

- followlinks：可选参数，默认为False。表示是否要遍历符号链接所指向的目录。

walk函数会返回一个生成器对象，通过遍历该对象可以获取到每个目录的路径、包含的子目录列表以及其中的文件列表。

下面是一个示例，演示如何使用os模块中的walk函数遍历目录树并打印出目录和文件的路径：

import os

def walk_directory(directory):
    for root, dirs, files in os.walk(directory):
        print('当前目录:', root)

        print('子目录列表:')
        for d in dirs:
            print(os.path.join(root, d))

        print('文件列表:')
        for file in files:
            print(os.path.join(root, file))
        print('===')

# 遍历当前目录
walk_directory('.')

运行以上代码，会从当前目录开始遍历目录树，并打印出每个目录的路径、包含的子目录列表和文件列表。

需要注意的是，walk函数会自动遍历所有的子目录，所以不需要手动递归调用该函数。

此外，还可以通过对walk函数的返回值进行判断来确定是否要停止遍历。若需要停止遍历，只需在循环中使用break即可。

import os

def walk_directory(directory):
    for root, dirs, files in os.walk(directory):
        print('当前目录:', root)

        print('子目录列表:')
        for d in dirs:
            print(os.path.join(root, d))

        print('文件列表:')
        for file in files:
            print(os.path.join(root, file))
        print('===')

        # 在这里添加停止遍历的条件
        if 'stop.txt' in files:
            break

# 遍历当前目录
walk_directory('.')

以上示例中，当目录中存在名为stop.txt的文件时，遍历会停止。