Python 使用 Python 遍历目录

在本文中，我们将介绍如何使用Python遍历目录。遍历目录是指在一个目录下递归地查找所有的文件和子目录。通过遍历目录，我们可以获取目录下所有文件的信息，如文件名、文件大小、创建时间等。

阅读更多：Python 教程

使用os模块遍历目录

Python中的os模块提供了一些方法来进行目录遍历操作。我们可以使用os模块的listdir函数获取目录下所有的文件和子目录，并使用isdir函数判断是否为子目录。下面是一个使用os模块遍历目录的示例代码：

import os

def traverse_directory(directory):
    for filename in os.listdir(directory):
        filepath = os.path.join(directory, filename)
        if os.path.isdir(filepath):
            print("Directory:", filepath)
            traverse_directory(filepath)
        else:
            print("File:", filepath)

上述代码中，traverse_directory函数接受一个目录路径作为参数。首先使用os.listdir函数获取目录下所有文件和子目录的列表，然后通过os.path.join方法获取子目录的完整路径。接着，使用os.path.isdir函数判断当前路径是否为子目录，如果是子目录则调用traverse_directory进行递归遍历，否则输出当前路径为文件。通过递归调用，我们可以遍历目录下的所有文件和子目录。

使用上述示例代码，我们可以遍历指定目录下的所有文件和子目录。例如，我们有以下目录结构：

root/
    |--dir1/
    |      |--file1.txt
    |      |--file2.txt
    |
    |--dir2/
           |--file3.txt
           |--file4.txt

如果我们调用traverse_directory("root")，将输出以下内容：

Directory: root/dir1
File: root/dir1/file1.txt
File: root/dir1/file2.txt
Directory: root/dir2
File: root/dir2/file3.txt
File: root/dir2/file4.txt

使用os.walk遍历目录

除了使用os模块进行目录遍历外，Python还提供了更方便的方法来遍历目录，即使用os.walk函数。os.walk函数会递归地遍历目录下的所有文件和子目录，并返回一个生成器。下面是一个使用os.walk遍历目录的示例代码：

import os

def traverse_directory(directory):
    for root, dirs, files in os.walk(directory):
        for filename in files:
            filepath = os.path.join(root, filename)
            print("File:", filepath)
        for dir in dirs:
            dirpath = os.path.join(root, dir)
            print("Directory:", dirpath)

上述代码中，traverse_directory函数接受一个目录路径作为参数。使用os.walk函数遍历目录时，它会返回一个生成器，每次迭代返回一个包含当前目录路径、子目录列表和文件列表的元组。通过对返回的元组进行解构，我们可以获取当前目录的路径、子目录列表和文件列表。接着，我们可以使用os.path.join方法将当前目录路径和文件名或子目录名拼接成完整的路径，并进行输出。

与使用os模块遍历目录相比，使用os.walk函数更加简洁和方便。使用os.walk函数，我们无需手动判断当前路径是否为子目录，也无需进行递归调用，os.walk函数会自动处理这些操作。

使用上述示例代码，我们可以遍历指定目录下的所有文件和子目录。例如，我们有以下目录结构：

root/
    |--dir1/
    |      |--file1.txt
    |      |--file2.txt
    |
    |--dir2/
           |--file3.txt
           |--file4.txt

如果我们调用traverse_directory("root")，将输出以下内容：

File: root/dir1/file1.txt
File: root/dir1/file2.txt
File: root/dir2/file3.txt
File: root/dir2/file4.txt
Directory: root/dir1
Directory: root/dir2

总结

本文介绍了如何使用Python遍历目录。我们学习了使用os模块的listdir函数和isdir函数来遍历目录，并使用递归的方式实现了遍历子目录的功能。同时，我们还学习了使用os.walk函数来更方便地遍历目录，无需手动处理子目录和递归调用的问题。

在实际项目中，遍历目录是一个常见的操作。通过遍历目录，我们可以对目录下的文件进行批量处理、统计文件数量和大小等。掌握了遍历目录的方法，我们可以更高效地处理目录操作，并且减少了手动处理文件和子目录的工作量。