图片按照csv文件中的标签保存在不同文件夹下

图片按照csv文件中的标签保存在不同文件夹下，文件夹以标签名字命名。

原始文件路径： data-train

                                 -test

                                 -train.csv

filename	label
0.jpg	plastic
1.jpg	paper

import pandas as pd
import os
import shutil #用于移动文件

#打开表格文件并读取
f=open("./refuse_data/train.csv","rb") #输入表格所在路径+名称
list=pd.read_csv(f)

#进行分类
for i in ['paper','glass','plastic','cardboard','metal','trash']:
    os.mkdir(i)
    listnew=list[list["label"]==i]
    l=listnew["filename"].tolist()
    j=str(i)
    for each in l:
        print(each,j)
        shutil.copy('./refuse_data/train/'+each,j)