📅  最后修改于: 2023-12-03 14:53:20.466000             🧑  作者: Mango
在Python中,我们经常需要处理大量的图像,如何遍历一个文件夹中的所有图像是一个常见的问题。下面介绍一种比较常见的方法。
os
是Python中的一个系统库,我们可以使用它来遍历文件夹。具体步骤如下:
import os
os.listdir()
获取指定文件夹中的所有文件名dir_path = 'path/to/directory'
file_list = os.listdir(dir_path)
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
用来拼接出文件的完整路径。
from PIL import Image
for image_path in image_list:
image = Image.open(image_path)
# 对图像进行操作
在这里我们使用Pillow
库来读取图像文件,Image.open
函数可以打开图像文件并返回一个Image
对象,我们可以通过这个对象对图像进行操作。比如调整大小、裁剪、旋转等。
使用os
模块遍历文件夹中的图像文件有一定的局限性,只能筛选特定后缀的文件,而且不支持递归遍历子文件夹中的文件。如果需要更高级的文件遍历功能,可以考虑使用os.walk()
或glob
模块。