python读取文件夹下的图片

Python读取文件夹下的图片

在实际的工作和项目中，我们经常需要处理大量的图片数据。而这些图片可能散落在不同的文件夹中，如果手动一个一个去读取，不仅费时费力，还容易出错。所以，使用Python来读取文件夹下的图片是一个非常实用的技巧。本文将介绍如何使用Python来读取文件夹下的图片，并提供相应的代码示例。

读取文件夹下的图片

首先，我们需要明确目标，即我们希望读取哪个文件夹下的图片。假设我们的图片文件夹路径为image_folder。

安装依赖库

在使用Python读取文件夹下的图片之前，我们需要安装PIL库（Python Imaging Library），它是Python中最常用的图像处理库之一。如果你还没有安装PIL库，可以使用以下命令进行安装：

pip install pillow

导入依赖库

安装完PIL库后，我们需要导入相应的依赖库：

from PIL import Image
import os

读取文件夹下的图片

下面是读取文件夹下的图片的代码示例：

def read_images_from_folder(folder):
    images = []
    for filename in os.listdir(folder):
        if filename.endswith(".jpg") or filename.endswith(".png"):
            img = Image.open(os.path.join(folder, filename))
            images.append(img)
    return images

代码中的read_images_from_folder函数接受一个文件夹路径作为输入参数，然后遍历文件夹下的所有文件名。如果文件名以.jpg或.png结尾，就使用PIL库中的Image.open方法打开图片，并将其添加到images列表中。最后，返回所有图片的列表。

使用示例

下面是一个使用示例，演示了如何读取文件夹下的图片并展示图片的尺寸：

image_folder = "path/to/folder"

images = read_images_from_folder(image_folder)

for img in images:
    print(img.size)

在上面的代码中，你需要将path/to/folder替换为你的图片文件夹路径。然后，调用read_images_from_folder函数读取图片，并使用img.size属性打印出每张图片的尺寸。

状态图

下面是一个使用mermaid语法绘制的状态图，用于说明上述代码的执行过程：

stateDiagram
    [*] --> Read_Folder
    Read_Folder --> Check_Extension
    Check_Extension --> Read_Image
    Read_Image --> Check_Extension
    Check_Extension --> Done
    Done --> [*]

上述状态图展示了代码的执行过程。首先，程序进入Read_Folder状态，然后判断文件扩展名的条件。如果满足条件，则进入Read_Image状态；否则，返回Check_Extension状态。当所有图片都读取完毕后，程序进入Done状态，然后返回到初始状态[*]。

总结

使用Python读取文件夹下的图片是一个非常实用的技巧，可以帮助我们快速有效地处理大量的图片数据。通过本文的介绍，你学会了如何使用Python来读取文件夹下的图片，并了解了相应的代码示例。希望本文能够对你在实际工作中处理图片数据时有所帮助。

参考资料

• [Pillow Documentation](
• [Python os module](