python 一个文件夹的图片平均分成多个文件夹

最新推荐文章于 2024-10-02 10:53:34 发布

XJBC

最新推荐文章于 2024-10-02 10:53:34 发布

阅读量75

点赞数

文章标签： python 开发语言

python相关学习资料：

https://edu.51cto.com/video/1158.html

https://edu.51cto.com/video/3832.html

https://edu.51cto.com/video/3502.html

Python实现图片自动分类存储

在处理大量图片数据时，将图片按照一定的规则自动分类存储到不同的文件夹中，可以大大提高数据管理的效率。本文将介绍如何使用Python实现这一功能。

背景介绍

在数据科学、机器学习等领域，我们经常需要处理大量的图片数据。如果所有图片都存储在一个文件夹中，查找和管理起来将非常不便。因此，将图片按照一定的规则自动分类存储到不同的文件夹中，可以大大提高工作效率。

功能需求

读取指定文件夹中的所有图片。
根据图片的数量，将图片平均分配到多个文件夹中。
如果图片数量不能整除，将多出来的图片分配到第一个文件夹中。

技术选型

为了实现上述功能，我们选择使用Python语言，主要原因有：

Python语法简洁，易于编写和理解。
Python有丰富的库支持，如os库用于文件操作，PIL库用于图片处理。

实现步骤

1. 读取图片

首先，我们需要读取指定文件夹中的所有图片。可以使用os库的os.listdir()函数遍历文件夹中的所有文件。

2. 分配文件夹

根据图片的数量，我们可以计算出每个文件夹应该存放的图片数量。如果图片数量不能整除，将多出来的图片分配到第一个文件夹中。

3. 移动图片

最后，我们需要将图片移动到对应的文件夹中。可以使用shutil库的shutil.move()函数实现。

代码实现

import os
import shutil

def classify_images(src_folder, dest_folder, num_folders):
    # 读取图片
    images = [f for f in os.listdir(src_folder) if f.endswith(('.png', '.jpg', '.jpeg'))]
    
    # 计算每个文件夹应该存放的图片数量
    images_per_folder = len(images) // num_folders
    remainder = len(images) % num_folders
    
    # 移动图片
    for i in range(num_folders):
        folder_name = os.path.join(dest_folder, f'folder_{i+1}')
        os.makedirs(folder_name, exist_ok=True)
        
        start = i * images_per_folder + (i if i < remainder else remainder)
        end = start + images_per_folder + (1 if i < remainder else 0)
        
        for j in range(start, end):
            shutil.move(os.path.join(src_folder, images[j]), folder_name)

# 使用示例
src_folder = 'path/to/source/folder'
dest_folder = 'path/to/destination/folder'
num_folders = 5
classify_images(src_folder, dest_folder, num_folders)