import pandas as pd
import os
import shutil
#读取文件
file=open("/Volumes/creazy_boy/kt/train1.csv","rb")
list=pd.read_csv(file)
list["FILE_ID"]=list["data"]
#创建文件夹,进行分类
for i in range(1,4):
label_dir = os.path.join('A'+"{:02}".format(i))
if not os.path.exists(label_dir):
os.mkdir(label_dir)
listnew=list[list["label"]==i]
list_image=listnew["FILE_ID"].tolist()
for img in list_image:
shutil.copy("/Volumes/creazy_boy/kt/train/"+img,label_dir)
python下根据csv将数据集按照类别分文件夹放置
最新推荐文章于 2022-09-12 15:04:11 发布