python下根据csv将数据集按照类别分文件夹放置

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)