读取csv内容，根据csv中的图片标签将图片放到不同文件夹下

1. 读取一个csv文件，预处理

def dataframe_creation():
    df4 = pd.read_csv('**.csv')
    df4['filename']='图片存放位置/'+df4['filename']
    return df4
df = dataframe_creation()

这一步是为了去除标签中没有相应图像的标签，可选

listImg = os.listdir('图片存放位置')
string = '图片存放位置/'
list2 = list(map(lambda orig_string: string + orig_string , listImg))
indexify =[]
for i in df.index:
    if df.iloc[i]['Left-Fundus'] in list2 and df.iloc[i]['Right-Fundus'] in list2:
        continue
    else:
        indexify.append(i)

df = df.drop(indexify)

2.分文件夹存放

import pandas as pd
import os
import shutil

# 读取文件
file = open("E:/Classify/Ocular_disease_recoginition_data/8_image/name11.csv", "rb")
list = pd.read_csv(file)

# 创建文件夹，进行分类
for i in range(0, 7):
    label_dir = os.path.join('A' + "{:02}".format(i))
    if not os.path.exists(label_dir):
        os.mkdir(label_dir)
    listnew = list[list["label_ID"] == i]#标签值
    list_image = listnew["filename"].tolist()#图像位置
    for img in list_image:
        shutil.copy("存放路径，若不填就放到py执行文件的文件夹下",img, label_dir)

label_ID和filename的格式如下：