python---ISIC2018预处理---根据类别文件存储位置复制

python—预处理—根据类别文件存储位置复制

最近在学习卷积神经网络，第一步做标签环节。数据集采用的是ISIC-2018数据，由于所以数据存储在一个文件夹里如下图所示。

其分类信息保存在.csv文件中，如下图所示

我想要根据.csv文件的类别信息分别存储到不同的文件夹下，如下图所示。

import pandas as pd
import os
import shutil


df=pd.read_csv('E:/ISIC2018_Task3/ISIC2018_Task3_Training_GroundTruth/ISIC2018_Task3_Training_GroundTruth.csv')  #.CSV文件位置
#print(df.head(10000))
#for i in range(0,10000):
#    print (i)
cwd='E:/ISIC2018_Task3/ISIC2018_Task3_Training_Input/' #图片位置
names = []
names = os.listdir(cwd)
print(names[4])
for i in range(0,10000):
    dfdata=[]
    dfdata=df.loc[i]
    if dfdata[1]==1:
        print("MEL")
        print(dfdata[0]+".jpg")
        a = dfdata[0]+".jpg"
        for j in range(0,10000):
            if names[j] == a:
               b="E:/ISIC2018_Task3/ISIC2018_Task3_Training_Input/"+names[j]
               print(b)
               shutil.copy(b, 'D:/DATA/MEL/')  # 复制
    if dfdata[2]==1:
        print("NV")
        print(dfdata[0]+".jpg")
        a = dfdata[0]+".jpg"
        for j in range(0,10000):
            if names[j] == a:
               b="E:/ISIC2018_Task3/ISIC2018_Task3_Training_Input/"+names[j]
               print(b)
               shutil.copy(b, 'D:/DATA/NV/')  # 复制
    if dfdata[3]==1:
        print("BCC")
        print(dfdata[0]+".jpg")
        a = dfdata[0]+".jpg"
        for j in range(0,10000):
            if names[j] == a:
                b="E:/ISIC2018_Task3/ISIC2018_Task3_Training_Input/"+names[j]
                print(b)
                shutil.copy(b, 'D:/DATA/BCC/')  # 复制
    if dfdata[4]==1:
        print("AKIEC")
        print(dfdata[0]+".jpg")
        a = dfdata[0]+".jpg"
        for j in range(0,10000):
            if names[j] == a:
                b="E:/ISIC2018_Task3/ISIC2018_Task3_Training_Input/"+names[j]
                print(b)
                shutil.copy(b, 'D:/DATA/AKIEC/')  # 复制
    if dfdata[5]==1:
        print("BKL")
        print(dfdata[0]+".jpg")
        a = dfdata[0]+".jpg"
        for j in range(0,10000):
            if names[j] == a:
                b="E:/ISIC2018_Task3/ISIC2018_Task3_Training_Input/"+names[j]
                print(b)
                shutil.copy(b, 'D:/DATA/BKL/')  # 复制
    if dfdata[6]==1:
        print("DF")
        print(dfdata[0]+".jpg")
        a = dfdata[0]+".jpg"
        for j in range(0,10000):
            if names[j] == a:
                b="E:/ISIC2018_Task3/ISIC2018_Task3_Training_Input/"+names[j]
                print(b)
                shutil.copy(b, 'D:/DATA/DF/')  # 复制
    if dfdata[7]==1:
        print("VASC")
        print(dfdata[0]+".jpg")
        a = dfdata[0]+".jpg"
        for j in range(0,10000):
            if names[j] == a:
                b="E:/ISIC2018_Task3/ISIC2018_Task3_Training_Input/"+names[j]
                print(b)
                shutil.copy(b, 'D:/DATA/VASC/')  # 复制