一文弄懂 Python os.walk()，轻松搞定文件处理和目录遍历

在 Python 中，处理文件和目录操作是常见的需求，特别是在数据处理、文件管理等场景中，了解如何高效地遍历目录结构是非常重要的。os 模块提供了许多与文件和目录相关的操作，其中 os.walk() 是一个非常有用的工具，能够轻松实现目录的遍历。

os.walk() 函数概述

os.walk() 是一个生成器，用于遍历指定目录及其子目录中的所有文件和文件夹。它返回一个三元组 (dirpath, dirnames, filenames)，其中：

• dirpath：当前遍历到的目录路径
• dirnames：该目录下的子目录名称列表
• filenames：该目录下的文件名称列表

通过使用 os.walk()，我们可以轻松地处理目录树中的每一个文件和文件夹。

基本用法

以下是一个简单的示例，展示了如何使用 os.walk() 遍历一个目录及其子目录，并打印出所有文件的完整路径：

import os

def list_files(starting_directory):
    for dirpath, dirnames, filenames in os.walk(starting_directory):
        print(f"当前目录: {dirpath}")
        for filename in filenames:
            print(f"文件: {os.path.join(dirpath, filename)}")

# 使用示例
starting_directory = '/path/to/your/directory'  # 替换为你的目录路径
list_files(starting_directory)

在这个示例中，我们定义了 list_files 函数，该函数接受一个目录路径作为参数。os.walk() 会遍历该目录，并打印出每个目录的路径以及其中的文件。

处理特定文件类型

在实际应用中，我们可能只对某些特定类型的文件感兴趣，例如文本文件或图像文件。我们可以在循环中添加条件判断来筛选文件类型：

def list_text_files(starting_directory):
    for dirpath, dirnames, filenames in os.walk(starting_directory):
        for filename in filenames:
            if filename.endswith('.txt'):  # 仅处理以 .txt 结尾的文件
                print(f"文本文件: {os.path.join(dirpath, filename)}")

# 使用示例
list_text_files(starting_directory)

在这个修改后的版本中，list_text_files 函数只打印以 .txt 结尾的文件路径，帮助用户快速定位所需的文件类型。

统计文件数量

os.walk() 还可以用来统计某个目录下的文件数量。以下是一个示例，统计目录及其子目录中的所有文件总数：

def count_files(starting_directory):
    total_files = 0
    for dirpath, dirnames, filenames in os.walk(starting_directory):
        total_files += len(filenames)
    print(f"总文件数量: {total_files}")

# 使用示例
count_files(starting_directory)

在这个示例中，我们定义了 count_files 函数，它遍历指定目录并统计文件数量，然后输出结果。

小结

os.walk() 是 Python 中非常强大的一个工具，能够帮助我们高效地遍历目录及其子目录，处理文件数据。通过一些简单的条件判断，我们可以快速筛选出我们所需的文件类型，或者进行文件统计等操作。无论是数据分析、文件管理还是开发维护，这个函数都能带来极大的便利。希望本文能帮助你更好地理解和使用 os.walk()。