Python遍历目录树中的函数调用的实际操作步骤简介(Python遍历目录树函数调用实战步骤详解)

原创

ithorizon 7个月前 (10-19) 阅读数 28 #后端开发

Python遍历目录树函数调用实战步骤详解

一、Python遍历目录树的概念

在Python中，遍历目录树指的是递归地访问一个目录及其所有子目录中的文件和文件夹。这种操作在处理文件系统时非常常见，比如统计文件大小、查找特定文件等。

二、使用os模块遍历目录树

Python的os模块提供了遍历目录树的方法。下面我们将详细介绍怎样使用os模块中的os.walk()函数来实现目录树的遍历。

2.1 导入os模块

首先，我们需要导入os模块。


import os

2.2 os.walk()函数的基本使用

os.walk()函数返回一个三元组(root, dirs, files)。

root：当前正在遍历的目录的路径。

dirs：当前目录下所有子目录的列表（不包括子目录下的目录）。

files：当前目录下所有非目录文件的列表。

下面是一个简洁的使用os.walk()遍历目录树的例子。


import os
def traverse_directory(directory):
    for root, dirs, files in os.walk(directory):
        print(f"当前目录: {root}")
        for file in files:
            print(f"文件: {file}")
        for dir in dirs:
            print(f"子目录: {dir}")
traverse_directory("/path/to/directory")

三、os.walk()的高级用法

os.walk()提供了topdown和bottomup参数，可以控制遍历的顺序。

3.1 topdown参数

topdown参数默认为True，意味着遍历会从上到下进行。在这种情况下，os.walk()会先访问父目录，然后是子目录。


import os
def traverse_directory_topdown(directory):
    for root, dirs, files in os.walk(directory, topdown=True):
        print(f"当前目录: {root}")
        for file in files:
            print(f"文件: {file}")
        for dir in dirs:
            print(f"子目录: {dir}")
traverse_directory_topdown("/path/to/directory")

3.2 bottomup参数

bottomup参数为True时，遍历会从下到上进行。这意味着os.walk()会先访问最底层的目录，然后逐级向上。


import os
def traverse_directory_bottomup(directory):
    for root, dirs, files in os.walk(directory, topdown=False):
        print(f"当前目录: {root}")
        for file in files:
            print(f"文件: {file}")
        for dir in dirs:
            print(f"子目录: {dir}")
traverse_directory_bottomup("/path/to/directory")

四、自定义遍历逻辑

在实际应用中，我们也许需要依文件或目录的特定属性来执行不同的操作。我们可以通过修改os.walk()的遍历逻辑来实现这一点。

4.1 过滤文件

例如，我们也许只对特定扩展名的文件感兴趣，可以添加一个过滤条件。


import os
def traverse_directory_filter(directory, extension):
    for root, dirs, files in os.walk(directory):
        for file in files:
            if file.endswith(extension):
                print(f"文件: {os.path.join(root, file)}")
traverse_directory_filter("/path/to/directory", ".txt")

4.2 修改目录结构

我们还可以在遍历过程中修改目录结构，比如删除特定的子目录或文件。


import os
def traverse_directory_modify(directory):
    for root, dirs, files in os.walk(directory):
        for file in files:
            if file == "example.txt":
                os.remove(os.path.join(root, file))
        for dir in dirs:
            if dir == "old_directory":
                os.rmdir(os.path.join(root, dir))
traverse_directory_modify("/path/to/directory")

五、使用pathlib模块遍历目录树

Python 3.4以上版本还提供了pathlib模块，它提供了一个面向对象的方法来处理文件系统路径。下面我们看看怎样使用pathlib模块遍历目录树。

5.1 导入pathlib模块

首先，我们需要导入pathlib模块。


from pathlib import Path

5.2 使用Path对象的遍历方法

Path对象提供了几个方法来遍历目录树，比如iterdir()、glob()等。


from pathlib import Path
def traverse_directory_pathlib(directory):
    path = Path(directory)
    for p in path.rglob("*"):
        if p.is_file():
            print(f"文件: {p}")
        elif p.is_dir():
            print(f"目录: {p}")
traverse_directory_pathlib("/path/to/directory")