python循环遍历文件夹中的图片

原创

mob64ca12d5dd85 2024-02-25 04:27:39 ©著作权

文章标签 Python 遍历文件夹文件名 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12d5dd85的原创作品，请联系作者获取转载授权，否则将追究法律责任

使用Python循环遍历文件夹中的图片

在日常工作和生活中，我们经常需要处理文件夹中的图片，比如查找特定格式的图片、批量处理图片等。Python作为一种强大的编程语言，提供了各种库和工具来处理文件操作，包括遍历文件夹中的图片文件。本文将介绍如何使用Python循环遍历文件夹中的图片，并给出代码示例。

遍历文件夹中的图片

在Python中，我们可以使用os模块来进行文件和文件夹操作，使用PIL或opencv等库来处理图片文件。下面是一个简单的流程图，展示了如何使用Python循环遍历文件夹中的图片：

flowchart TD
    A(开始)
    B{检查文件夹}
    C{是否是图片文件}
    D(处理图片文件)
    E(结束)

    A --> B
    B --> C
    C -- 是 --> D
    C -- 否 --> B
    D --> E

代码示例

下面是一个示例代码，演示了如何使用Python循环遍历文件夹中的图片，并输出每个图片的文件名：

import os

# 设置图片文件格式
image_formats = ['.jpg', '.jpeg', '.png']

# 遍历文件夹
folder_path = 'path_to_folder'
for file_name in os.listdir(folder_path):
    if os.path.isfile(os.path.join(folder_path, file_name)):
        # 检查是否是图片文件
        if any(file_name.endswith(image_format) for image_format in image_formats):
            print(file_name)

上述代码中，首先定义了图片文件的格式为.jpg, .jpeg, .png，然后遍历指定文件夹中的所有文件，检查每个文件是否是图片文件，如果是则输出文件名。

序列图

为了更直观地展示上述代码的执行流程，我们可以使用序列图来描述：

sequenceDiagram
    participant Python
    participant OS
    participant File

    Python->>OS: 遍历文件夹
    OS->>File: 获取文件名
    File->>Python: 返回文件名
    Python->>File: 检查文件类型
    File->>Python: 返回文件类型
    Python-->>Python: 输出图片文件名

在序列图中，Python首先遍历文件夹，然后获取文件名，检查文件类型，最后输出图片文件名。