python 遍历文件夹

在日常工作中，我们经常需要读取文件夹中的文件列表，可能是为了做一些统计工作，也可能是为了对文件进行某些操作。Python 提供了一些库，可以帮助我们实现文件夹遍历。在本文中，我将介绍三种常用的遍历文件夹的方法，并且给出每种方法的使用示例和优缺点分析。

方法一：os.walk()

os.walk() 是 Python 中最常用的遍历文件夹的方法。这个方法会自动地递归遍历文件夹中的所有子文件夹。在遍历过程中，返回的是一个元组(root, dirs, files)，其中 root 表示当前遍历到的文件夹的路径，dirs 表示当前文件夹下的子文件夹，files 表示当前文件夹下的所有文件。

import os

# 遍历文件夹
def traversal_folder(folder_path):
    for root, dirs, files in os.walk(folder_path):
        # 遍历当前文件夹下的所有文件
        for file_name in files:
            file_path = os.path.join(root, file_name)
            print(file_path)

# 调用函数
folder_path = r"D:\data"
traversal_folder(folder_path)

优点：os.walk() 能够自动遍历文件夹下的子文件夹，无需手动递归。

缺点：对于大型文件夹，os.walk() 会占用大量的内存，因为它需要把所有的文件都存储在内存中，会导致程序崩溃。

方法二：os.listdir()

os.listdir() 是 Python 中另一种常用的遍历文件夹的方法。这个方法会返回指定路径下的所有文件和文件夹的列表，但是不会递归遍历。

import os

# 遍历文件夹
def traversal_folder(folder_path):
    for file_name in os.listdir(folder_path):
        file_path = os.path.join(folder_path, file_name)
        print(file_path)

# 调用函数
folder_path = r"D:\data"
traversal_folder(folder_path)

优点：os.listdir() 执行速度比 os.walk() 更快。

缺点：os.listdir() 不会递归遍历，如果需要递归遍历的话，还需要在函数中加入递归操作。

方法三：glob.glob()

glob.glob() 是 Python 中一种使用比较少见的遍历文件夹的方法。这个方法会根据指定的文件名模式匹配到的所有文件和文件夹，并以一个列表的形式返回。类似于 os.listdir()，不会递归遍历。

import glob

# 遍历文件夹
def traversal_folder(folder_path):
    for file_path in glob.glob(os.path.join(folder_path, '*')):
        print(file_path)

# 调用函数
folder_path = r"D:\data"
traversal_folder(folder_path)

优点：glob.glob() 使用简单，代码量少。

缺点：glob.glob() 不会递归遍历，如果需要递归遍历的话，还需要在函数中加入递归操作。