读取CSV内容，根据不同的图片标签放到指定文件夹

读取csv文件内容，根据不同的图片标签到指定文件夹。

原本的数据是这样的

然后根据CSV文件的内容

进行读取，根据不同的label将图片放到指定的label 文件夹里

import pandas as pd
import os
import shutil  

#读取文件
file=open("submission.csv","rb")  
list=pd.read_csv(file)
list["FILE_PNG"]=".png" 
list["FILE_ID"]=list["ID"]+list["FILE_PNG"] 

   

#创建文件夹，进行分类
for i in range(1,13):
    label_dir = os.path.join('A'+"{:02}".format(i))
    if not os.path.exists(label_dir):
        os.mkdir(label_dir)
    listnew=list[list["Pred_Label"]==i]
    list_image=listnew["FILE_ID"].tolist()
    for img in list_image:
        shutil.copy("/Pad/test/"+img,label_dir)

最后的结果是这样的

至此，就完成了将csv文件中，根据图片的不同标签进行归类，放到指定文件夹中