python如何遍历文件夹下的图像

使用Python遍历文件夹下的图像的方法有：os模块、glob模块、pathlib模块。 其中，os模块和glob模块是比较常用的方式，pathlib模块则是Python 3.4引入的新模块，提供了面向对象的路径操作。下面将详细介绍这几种方法。

一、使用os模块

os模块是Python标准库中的一部分，提供了一些与操作系统交互的函数。使用os模块遍历文件夹下的图像文件，可以通过os.listdir()函数列出目录中的所有文件和子目录，然后通过os.path.isfile()函数和字符串方法判断文件类型。

import os
def traverse_images_os(directory):
    for filename in os.listdir(directory):
        if filename.endswith('.jpg') or filename.endswith('.png') or filename.endswith('.jpeg'):
            print(os.path.join(directory, filename))
directory = '/path/to/your/directory'
traverse_images_os(directory)

在上面的代码中，首先导入os模块，然后定义了一个遍历函数traverse_images_os()，它接收一个目录路径作为参数。通过os.listdir()函数获取该目录下的所有文件和子目录，接着判断文件扩展名是否是图像格式，如果是，则打印文件的完整路径。

二、使用glob模块

glob模块提供了一个函数，用于在目录中查找符合特定模式的文件。相比于os模块，glob模块更简洁和易用。

import glob
def traverse_images_glob(directory):
    for filepath in glob.glob(os.path.join(directory, '*')):
        if filepath.endswith('.jpg') or filepath.endswith('.png') or filepath.endswith('.jpeg'):
            print(filepath)
directory = '/path/to/your/directory'
traverse_images_glob(directory)

在上面的代码中，首先导入glob模块，然后定义了一个遍历函数traverse_images_glob()。通过glob.glob()函数获取符合模式的文件列表，接着判断文件扩展名是否是图像格式，如果是，则打印文件的完整路径。

三、使用pathlib模块

pathlib模块是Python 3.4引入的新模块，提供了面向对象的路径操作。相比于os模块，pathlib模块的代码更加简洁和易读。

from pathlib import Path
def traverse_images_pathlib(directory):
    p = Path(directory)
    for filepath in p.glob('*'):
        if filepath.suffix in ['.jpg', '.png', '.jpeg']:
            print(filepath)
directory = '/path/to/your/directory'
traverse_images_pathlib(directory)

在上面的代码中，首先从pathlib模块中导入Path类，然后定义了一个遍历函数traverse_images_pathlib()。通过创建Path对象，使用Path.glob()方法获取符合模式的文件列表，接着判断文件扩展名是否是图像格式，如果是，则打印文件的完整路径。

四、递归遍历子目录

有时候我们需要遍历目录及其子目录中的所有图像文件，可以通过递归的方式实现。以下是使用os模块递归遍历的示例：

import os
def traverse_images_recursive(directory):
    for root, dirs, files in os.walk(directory):
        for filename in files:
            if filename.endswith('.jpg') or filename.endswith('.png') or filename.endswith('.jpeg'):
                print(os.path.join(root, filename))
directory = '/path/to/your/directory'
traverse_images_recursive(directory)

在上面的代码中，通过os.walk()函数遍历目录及其子目录。os.walk()函数返回一个三元组(root, dirs, files)，其中root是当前目录路径，dirs是子目录列表，files是文件列表。接着判断文件扩展名是否是图像格式，如果是，则打印文件的完整路径。

五、性能优化与扩展

在实际应用中，可能需要处理大量的图像文件，这时可以考虑一些性能优化和扩展的方法。比如：

使用多线程或多进程：可以使用Python的多线程或多进程库（如threading、multiprocessing）来加速遍历和处理图像文件的过程。
过滤特定图像格式：可以通过使用正则表达式或更高级的文件过滤库（如fnmatch）来更灵活地过滤图像文件。
处理图像文件：在遍历图像文件时，可能需要对图像文件进行一些处理（如读取、裁剪、转换格式等），可以使用Pillow库或OpenCV库来处理图像文件。

以下是一个使用多线程和Pillow库读取图像文件的示例：

import os
from PIL import Image
import threading
def process_image(filepath):
    try:
        with Image.open(filepath) as img:
            img = img.convert('RGB')
            # 进行一些图像处理操作
            print(f'Processed: {filepath}')
    except Exception as e:
        print(f'Error processing {filepath}: {e}')
def traverse_images_multithread(directory):
    threads = []
    for root, dirs, files in os.walk(directory):
        for filename in files:
            if filename.endswith('.jpg') or filename.endswith('.png') or filename.endswith('.jpeg'):
                filepath = os.path.join(root, filename)
                thread = threading.Thread(target=process_image, args=(filepath,))
                threads.append(thread)
                thread.start()
    for thread in threads:
        thread.join()
directory = '/path/to/your/directory'
traverse_images_multithread(directory)