Python 从目录中导入图片到列表或字典

在本文中，我们将介绍如何使用Python从指定目录中导入图片，并将其保存到列表或字典中。导入图片可以帮助我们在处理图像数据、图像识别、计算机视觉等方面的项目中更加便捷地管理和处理图片。

阅读更多：Python 教程

导入必要的库

在开始之前，我们需要导入一些必要的库来处理图片和目录操作。我们将使用Pillow库来处理图片，使用os库来操作文件和目录。

from PIL import Image

import os

从目录中导入图片到列表

首先，我们将演示如何将目录中的所有图片导入到一个列表中。这对于简单的图片处理操作非常方便。以下是一个示例代码：

# 设置目录路径

directory = '/path/to/directory'

# 遍历目录中的所有文件

for filename in os.listdir(directory):

# 检查文件是否是图片文件

if filename.endswith(".jpg") or filename.endswith(".png"):

# 使用Pillow库打开并处理图片

image = Image.open(os.path.join(directory, filename))

# 添加图片到列表中

image_list.append(image)

在上述代码中，我们首先设置了目录的路径。然后，我们使用os.listdir()函数遍历目录中的所有文件。通过检查文件的扩展名，我们确定文件是一个图片文件。然后，我们使用Pillow库的Image.open()函数打开图片，并将其添加到图片列表中。

从目录中导入图片到字典

除了将图片导入到列表中，我们还可以将图片导入到字典中，以便更好地对图片进行管理和访问。以下是一个示例代码：

# 设置目录路径

directory = '/path/to/directory'

# 创建空字典

image_dict = {}

# 遍历目录中的所有文件

for filename in os.listdir(directory):

# 检查文件是否是图片文件

if filename.endswith(".jpg") or filename.endswith(".png"):

# 使用Pillow库打开并处理图片

image = Image.open(os.path.join(directory, filename))

# 添加图片到字典中，以文件名作为键

image_dict[filename] = image

在上述代码中，我们首先设置了目录的路径。然后，我们创建了一个空字典来存储图片。通过遍历目录中的所有文件，并通过检查文件的扩展名来确定文件是一个图片文件。然后，我们使用Pillow库的Image.open()函数打开图片，并将其添加到字典中，以文件名作为键。

总结

在本文中，我们学习了如何使用Python从指定目录中导入图片，并将其保存到列表或字典中。通过这种方式，我们可以更方便地处理图片数据，并在图像处理和计算机视觉等项目中应用。希望本文对您学习Python处理图片有所帮助！