读取CSV内容,根据不同的图片标签放到指定文件夹
生活随笔
收集整理的這篇文章主要介紹了
读取CSV内容,根据不同的图片标签放到指定文件夹
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
讀取csv文件內容,根據不同的圖片標簽到指定文件夾。
原本的數據是這樣的
然后根據CSV文件的內容
進行讀取,根據不同的label將圖片放到指定的label 文件夾里
import pandas as pd import os import shutil #讀取文件 file=open("submission.csv","rb") list=pd.read_csv(file) list["FILE_PNG"]=".png" list["FILE_ID"]=list["ID"]+list["FILE_PNG"] #創建文件夾,進行分類 for i in range(1,13):label_dir = os.path.join('A'+"{:02}".format(i))if not os.path.exists(label_dir):os.mkdir(label_dir)listnew=list[list["Pred_Label"]==i]list_image=listnew["FILE_ID"].tolist()for img in list_image:shutil.copy("/Pad/test/"+img,label_dir)最后的結果是這樣的
至此,就完成了將csv文件中,根據圖片的不同標簽進行歸類,放到指定文件夾中
總結
以上是生活随笔為你收集整理的读取CSV内容,根据不同的图片标签放到指定文件夹的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: python 打开文件,读取文件内容
- 下一篇: Linux网络管理实 验 指 导