使用 Python 获取文件夹下的所有图片
在软件开发过程中，能够有效地操作和处理文件系统是非常重要的技能之一。尤其是在图像处理或数据分析领域，获取特定类型的文件（如图片）常常是任务的第一步。本文将指导你如何使用 Python 获取指定文件夹下的所有图片。
整体流程
接下来，我们将通过以下几个步骤来实现目标：
| 步骤 | 描述 |
|---|---|
| 1 | 导入必要的库 |
| 2 | 定义目标文件夹路径 |
| 3 | 获取文件夹内的文件 |
| 4 | 判断文件类型 |
| 5 | 输出图片文件列表 |
每一步的实现详细说明
1. 导入必要的库
首先，我们需要一些库来帮助我们完成任务。这里我们会使用 os 和 fnmatch 库。
# 导入 os 库，用于操作文件和目录
import os
# 导入 fnmatch 库，用于匹配文件名
import fnmatch
2. 定义目标文件夹路径
接下来，我们定义想要获取图片的文件夹路径。
# 定义文件夹路径，可以根据需要修改
folder_path = 'path/to/your/folder'
3. 获取文件夹内的文件
我们使用 os.listdir 方法来获取指定文件夹内的所有文件和目录。
# 获取文件夹内的所有文件和目录
all_files = os.listdir(folder_path)
4. 判断文件类型
在此步骤中，我们将通过文件扩展名来过滤出所有图片文件。
# 初始化一个空列表来存储图片文件
image_files = []
# 遍历文件夹中的所有文件
for filename in all_files:
# 使用 fnmatch 库匹配特定扩展名的图片文件
if fnmatch.fnmatch(filename, '*.jpg') or fnmatch.fnmatch(filename, '*.png') or fnmatch.fnmatch(filename, '*.jpeg'):
# 如果匹配，则将文件路径添加到列表中
image_files.append(os.path.join(folder_path, filename))
5. 输出图片文件列表
最后，我们可以输出我们找到的所有图片文件的路径。
# 输出找到的图片文件路径
for img in image_files:
print(img)
整合全部代码
将上述代码整合在一起，可得出完整代码如下：
import os
import fnmatch
folder_path = 'path/to/your/folder'
all_files = os.listdir(folder_path)
image_files = []
for filename in all_files:
if fnmatch.fnmatch(filename, '*.jpg') or fnmatch.fnmatch(filename, '*.png') or fnmatch.fnmatch(filename, '*.jpeg'):
image_files.append(os.path.join(folder_path, filename))
for img in image_files:
print(img)
序列图
以下是一个简单的序列图，展示了这个过程的步骤。
sequenceDiagram
participant User
participant Python
User->>Python: 导入库
User->>Python: 定义文件夹路径
User->>Python: 获取文件
Python-->>User: 返回文件夹中的文件
User->>Python: 判断文件类型
Python-->>User: 输出图片文件列表
旅行图
另外，我们可以用旅行图更清晰地展示这个过程：
journey
title 获取文件夹下的所有图片
section 导入库
导入 os 和 fnmatch 库: 5: User
section 定义文件夹路径
定义目标文件夹: 4: User
section 获取文件夹内的文件
列出所有文件: 4: User
section 过滤图片文件
判断文件扩展名: 4: User
section 输出结果
打印图片文件: 5: User
结尾
通过以上的步骤，你已经学会了如何使用 Python 获取指定文件夹中的所有图片文件。这个过程展示了如何利用 Python 的标准库来处理文件和文件夹的基本操作。今后，你可以利用这个基础，扩展更多功能，如对图片进行处理、分析等。希望这篇文章对你的学习有所帮助！如有任何疑问，欢迎提问。
