import os
import shutil
import pandas as pd
import random

# 打开表格文件并读取
f = open("E:train1/trainLabels.csv", "rb")  # 打开csv文件
list = pd.read_csv(f)  # 这句不能少
# print(list)
# 创建文件夹
for i in range(3):  # "6"指的是0-5总共6个类别
    if not os.path.exists('E:/train1/' + str(i)):  # 最后一个 / 不要漏
        os.mkdir('E:/train1/' + str(i))
# 进行分类
for i in range(3):

    listnew = list[list["level"] == i]  # 对应csv文件标签 那一栏的标题

    l = listnew["image"].tolist()  # 对应csv文件图片那一栏的标题

    j = str(i)
    for each in l:
        shutil.move('E:/train1/train/' + each +'.jpeg', 'E:/train1/' + j)
print("完成")

亲测可以跑通执行

Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐