📜  如何遍历文件夹Python中的图像?(1)

📅  最后修改于: 2023-12-03 14:53:20.466000             🧑  作者: Mango

如何遍历文件夹Python中的图像?

在Python中,我们经常需要处理大量的图像,如何遍历一个文件夹中的所有图像是一个常见的问题。下面介绍一种比较常见的方法。

使用os模块遍历文件夹

os是Python中的一个系统库,我们可以使用它来遍历文件夹。具体步骤如下:

  1. 导入os模块
import os
  1. 使用os.listdir()获取指定文件夹中的所有文件名
dir_path = 'path/to/directory'
file_list = os.listdir(dir_path)
  1. 遍历文件列表并筛选出后缀为图片格式的文件
image_list = []
for file_name in file_list:
    if file_name.endswith('.jpg') or file_name.endswith('.png'):
        image_list.append(os.path.join(dir_path, file_name))

在这里我们假设我们的目标文件夹中只包含.jpg.png格式的图像文件,所以在遍历时我们筛选出这两种格式的文件并加入到image_list中。os.path.join用来拼接出文件的完整路径。

  1. 使用Pillow库读取图像文件
from PIL import Image

for image_path in image_list:
    image = Image.open(image_path)
    # 对图像进行操作

在这里我们使用Pillow库来读取图像文件,Image.open函数可以打开图像文件并返回一个Image对象,我们可以通过这个对象对图像进行操作。比如调整大小、裁剪、旋转等。

总结

使用os模块遍历文件夹中的图像文件有一定的局限性,只能筛选特定后缀的文件,而且不支持递归遍历子文件夹中的文件。如果需要更高级的文件遍历功能,可以考虑使用os.walk()glob模块。