Python os.walk控制遍历深度

在Python中，可以使用os.walk函数来遍历目录结构。os.walk函数会递归地遍历目录结构，包括所有的子目录和文件。如果需要控制遍历深度，可以使用os.walk函数的depth参数来限制遍历的深度。

depth参数表示遍历的深度限制。如果depth参数为None（默认值），则os.walk函数将遍历整个目录结构。如果depth参数为正整数，则os.walk函数将在遍历到指定深度后停止遍历子目录。例如，如果depth参数为1，则os.walk函数将只遍历当前目录，而不会递归遍历子目录。如果depth参数为0，则os.walk函数将不会遍历任何目录。

以下是一个示例代码，演示如何使用os.walk函数的depth参数来控制遍历深度：

import os

def walk_dir(root_dir, depth=None):
    for root, dirs, files in os.walk(root_dir):
        if depth is not None and root.count(os.sep) - root_dir.count(os.sep) >= depth:
            # 当前目录深度超过指定深度，不再继续递归
            continue
        # 处理当前目录下的文件
        for file in files:
            file_path = os.path.join(root, file)
            print(file_path)
        # 处理当前目录下的子目录
        for dir in dirs:
            dir_path = os.path.join(root, dir)
            print(dir_path)
            # 递归遍历子目录
            walk_dir(dir_path, depth)

在上述示例代码中，`depth`参数用于限制遍历深度。`root_dir`参数表示要遍历的根目录。在`for root, dirs, files in os.walk(root_dir):`这一行代码中，`os.walk`函数开始遍历目录结构。`root`表示当前目录的路径，`dirs`表示当前目录下的子目录列表，`files`表示当前目录下的文件列表。如果当前目录深度超过指定深度，则使用`continue`语句跳过当前目录，不再继续递归遍历。否则，会分别遍历当前目录下的文件和子目录，并递归遍历子目录（如果深度未超过指定深度）。