根据CSV文件中标签，将图片分别移动到各个文件夹 python

根据如图所示放置csv文件和图片文件夹，

修改好csv_path、picture_path、image、labels后运行代码即可

import csv
import os
import shutil 


csv_path="train.csv"#csv文件地址
picture_path="train_images/"#图片文件夹地址
image="image"#csv文件第一列名称
labels="labels"#csv文件第二列名称

with open(csv_path) as f:
    f_csv=csv.DictReader(f)
    for row in f_csv:
        if not os.path.exists(picture_path+row.get(labels)):
            os.mkdir(picture_path+row.get(labels))
        shutil.move(picture_path+row.get(image),picture_path+row.get(labels)+'/'+row.get(image)) #zp