按标注将图片分类

import pandas as pd
import os
import shutil

# 读取文件
file = open("D:\\Project\\test\\picture\\data.csv", "rb")
list = pd.read_csv(file)
list["FILE_ID"] = list["ID"]
root = "D:\\Project\\test\\picture"
# 创建文件夹，进行分类
for i in range(9):
    label_dir = os.path.join(root,"{:1}".format(i))
    if not os.path.exists(label_dir):
        os.mkdir(label_dir)
    listnew = list[list["label"] == i]
    print(listnew)
    list_image = listnew["FILE_ID"].tolist()
    print(list_image)
    for img in list_image:
        print(img)
        shutil.copy("D:\\Project\\test\\picture\\img\\" + img, label_dir)