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

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

在Python中遍历文件夹下的图片可以使用os模块、glob模块、pathlib模块等方法。这些方法允许你轻松地访问和处理文件夹中的所有图片文件。使用os模块、使用glob模块、使用pathlib模块。以下是其中一个方法的详细描述。

使用os模块：os模块是Python标准库的一部分，提供了与操作系统进行交互的功能。使用os模块可以遍历文件夹中的所有文件，并通过文件扩展名来筛选出图片文件。以下是一个示例代码：

import os
def traverse_images(folder_path):
    images = []
    for root, dirs, files in os.walk(folder_path):
        for file in files:
            if file.endswith(('.png', '.jpg', '.jpeg', '.gif', '.bmp')):
                images.append(os.path.join(root, file))
    return images
folder_path = 'path/to/your/folder'
image_files = traverse_images(folder_path)
print(image_files)

一、使用os模块

os模块是Python标准库的一部分，提供了与操作系统进行交互的功能。使用os模块可以遍历文件夹中的所有文件，并通过文件扩展名来筛选出图片文件。

1.1 os.walk()方法

os.walk()方法生成目录树下的所有文件名，它是一个生成器，生成的是一个三元组(root, dirs, files)。

import os
def traverse_images(folder_path):
    images = []
    for root, dirs, files in os.walk(folder_path):
        for file in files:
            if file.endswith(('.png', '.jpg', '.jpeg', '.gif', '.bmp')):
                images.append(os.path.join(root, file))
    return images
folder_path = 'path/to/your/folder'
image_files = traverse_images(folder_path)
print(image_files)

在这个例子中，os.walk()会递归地遍历指定文件夹中的所有文件和子文件夹。通过检查文件名的扩展名，可以筛选出图片文件，并将它们的路径存储在一个列表中。

1.2 os.listdir()方法

os.listdir()方法返回指定文件夹中的所有文件和文件夹的名称列表。可以结合os.path.isdir()和os.path.isfile()方法来区分文件和文件夹。

import os
def traverse_images(folder_path):
    images = []
    for item in os.listdir(folder_path):
        item_path = os.path.join(folder_path, item)
        if os.path.isfile(item_path) and item.endswith(('.png', '.jpg', '.jpeg', '.gif', '.bmp')):
            images.append(item_path)
        elif os.path.isdir(item_path):
            images.extend(traverse_images(item_path))
    return images
folder_path = 'path/to/your/folder'
image_files = traverse_images(folder_path)
print(image_files)

在这个例子中，os.listdir()返回指定文件夹中的所有文件和文件夹的名称列表。通过判断每个项目是否为文件或文件夹，可以递归地遍历所有子文件夹，并筛选出图片文件。

二、使用glob模块

glob模块提供了一个函数，用于查找与指定模式匹配的路径名。它支持通配符匹配，可以方便地查找特定类型的文件。

2.1 glob.glob()方法

glob.glob()方法返回所有匹配的文件路径列表。可以使用通配符模式来匹配图片文件。

import glob
import os
def traverse_images(folder_path):
    image_files = []
    for ext in ['png', 'jpg', 'jpeg', 'gif', 'bmp']:
        image_files.extend(glob.glob(os.path.join(folder_path, '', f'*.{ext}'), recursive=True))
    return image_files
folder_path = 'path/to/your/folder'
image_files = traverse_images(folder_path)
print(image_files)

在这个例子中，glob.glob()方法使用通配符模式查找所有匹配的图片文件。通过设置recursive=True，可以递归地查找所有子文件夹中的图片文件。

2.2 glob.iglob()方法

glob.iglob()方法与glob.glob()方法类似，但它返回的是一个迭代器，而不是一个列表。对于大文件夹来说，这种方式更节省内存。

import glob
import os
def traverse_images(folder_path):
    image_files = []
    for ext in ['png', 'jpg', 'jpeg', 'gif', 'bmp']:
        for file_path in glob.iglob(os.path.join(folder_path, '', f'*.{ext}'), recursive=True):
            image_files.append(file_path)
    return image_files
folder_path = 'path/to/your/folder'
image_files = traverse_images(folder_path)
print(image_files)

在这个例子中，glob.iglob()方法返回一个迭代器，可以逐个获取匹配的图片文件路径。这样在处理大文件夹时，可以节省内存。

三、使用pathlib模块

pathlib模块是Python 3.4引入的面向对象的路径操作库，提供了更加简洁和直观的路径操作方法。

3.1 Path.glob()方法

Path.glob()方法返回一个生成器，生成匹配指定模式的路径。

from pathlib import Path
def traverse_images(folder_path):
    image_files = []
    folder = Path(folder_path)
    for ext in ['png', 'jpg', 'jpeg', 'gif', 'bmp']:
        image_files.extend(folder.glob(f'/*.{ext}'))
    return image_files
folder_path = 'path/to/your/folder'
image_files = traverse_images(folder_path)
print(image_files)

在这个例子中，Path.glob()方法返回一个生成器，可以生成匹配指定模式的图片文件路径。通过使用'/*.{ext}'模式，可以递归地查找所有子文件夹中的图片文件。

3.2 Path.rglob()方法

Path.rglob()方法是Path.glob()方法的递归版本，等价于使用'/*.{ext}'模式。

from pathlib import Path
def traverse_images(folder_path):
    image_files = []
    folder = Path(folder_path)
    for ext in ['png', 'jpg', 'jpeg', 'gif', 'bmp']:
        image_files.extend(folder.rglob(f'*.{ext}'))
    return image_files
folder_path = 'path/to/your/folder'
image_files = traverse_images(folder_path)
print(image_files)

在这个例子中，Path.rglob()方法返回一个生成器，可以递归地生成匹配指定模式的图片文件路径。

四、处理图片文件

遍历文件夹中的图片文件后，可以对这些图片文件进行进一步处理。例如，读取图片、显示图片、进行图像处理等。

4.1 使用Pillow库读取和显示图片

Pillow是Python的一个图像处理库，提供了图像读取、保存、显示和处理功能。

from PIL import Image
def process_images(image_files):
    for image_file in image_files:
        with Image.open(image_file) as img:
            img.show()
folder_path = 'path/to/your/folder'
image_files = traverse_images(folder_path)
process_images(image_files)

在这个例子中，使用Pillow库的Image.open()方法读取图片文件，并使用img.show()方法显示图片。

4.2 使用OpenCV库读取和显示图片

OpenCV是一个开源计算机视觉库，提供了丰富的图像处理功能。

import cv2
def process_images(image_files):
    for image_file in image_files:
        img = cv2.imread(image_file)
        cv2.imshow('Image', img)
        cv2.waitKey(0)
        cv2.destroyAllWindows()
folder_path = 'path/to/your/folder'
image_files = traverse_images(folder_path)
process_images(image_files)

在这个例子中，使用OpenCV库的cv2.imread()方法读取图片文件，并使用cv2.imshow()方法显示图片。使用cv2.waitKey()方法等待用户按键，使用cv2.destroyAllWindows()方法关闭显示窗口。

五、总结

本文介绍了在Python中遍历文件夹下的图片文件的三种方法：使用os模块、使用glob模块、使用pathlib模块。每种方法都有其优缺点，可以根据具体需求选择合适的方法。遍历图片文件后，可以使用Pillow库或OpenCV库对图片进行进一步处理。通过这些方法，可以方便地访问和处理文件夹中的所有图片文件，为后续的图像处理和分析奠定基础。