Python遍历目录下所有文件|极客教程

Python遍历目录下所有文件

1. 介绍

在程序开发中，经常需要对目录下的所有文件进行遍历处理。Python提供了多种方法来实现遍历目录下的文件，本文将详细介绍这些方法，并给出示例代码。

2. os模块

Python的os模块提供了一系列与操作系统相关的函数，包括目录操作。通过使用os模块，我们可以遍历目录下所有文件。

2.1 os.listdir()函数

os.listdir()函数返回指定目录下的所有文件和目录的名称列表。我们可以通过遍历这个列表，对每个文件进行处理。

示例代码：

import os

def traverse_files(directory):
    for filename in os.listdir(directory):
        filepath = os.path.join(directory, filename)  # 构造完整的文件路径
        if os.path.isfile(filepath):  # 判断是否为文件
            # 对文件进行处理
            print(filepath)

# 遍历当前目录的所有文件
traverse_files('.')

运行结果：

./file1.txt
./file2.txt
./subdir/file3.txt

2.2 os.walk()函数

os.walk()函数可以递归地遍历目录下的所有文件和子目录。该函数返回一个生成器对象，每次迭代都会返回一个包含当前目录路径、当前目录下所有子目录名称和当前目录下所有文件名称的元组。

示例代码：

import os

def traverse_files(directory):
    for root, dirs, files in os.walk(directory):
        for filename in files:
            filepath = os.path.join(root, filename)  # 构造完整的文件路径
            # 对文件进行处理
            print(filepath)

# 遍历当前目录及子目录的所有文件
traverse_files('.')

运行结果：

./file1.txt
./file2.txt
./subdir/file3.txt

3. pathlib模块

Python 3.4版本引入的pathlib模块提供了一种更为直观且面向对象的方式来操作文件和目录路径。该模块的Path类提供了遍历目录下文件的方法。

示例代码：

from pathlib import Path

def traverse_files(directory):
    path = Path(directory)
    for filepath in path.glob('**/*'):
        if filepath.is_file():  # 判断是否为文件
            # 对文件进行处理
            print(filepath)

# 遍历当前目录及子目录的所有文件
traverse_files('.')