当前位置： 移动技术网 > IT编程>脚本编程>Python > 使用python将图片按标签分入不同文件夹的方法

使用python将图片按标签分入不同文件夹的方法

2019年06月13日  | 移动技术网IT编程  | 我要评论

小黑论坛,风雪记牌器,烦恼歌英文版叫什么

给定图像集如下，所有类别的图片均在一个文件夹内：

python将图片按标签分入不同文件夹

给定与图片名相匹配的表格，声明每张图片对应的类别（共有20个类别）：

python将图片按标签分入不同文件夹

那么，如何根据表格中所给的类别将图片分入对应的文件夹内呢？以我的情况为例，我想将图片分为20类（category_id有0-19共20类），可利用下面的代码进行分类（经细心网友指正，代码已做出修改）。

#引入相关库

import pandas as pd
import os
import shutil #用于移动文件

#打开表格文件并读取
f=open("list.csv","rb") #输入表格所在路径+名称
list=pd.read_csv(f)
list["file_id_jpg"]=".jpg" #建立图片名与类别相对应
list["file_id1"]=list["file_id"]+list["file_id_jpg"] #建立图片名与类别相对应

#创建文件夹
for i in range(20):
 os.mkdir(str(i))

#进行分类
for i in range(0,20):
 listnew=list[list["category_id"]==i]
 l=listnew["file_id1"].tolist()
 j=str(i)
 for each in l:
  shutil.move(each,j)

大功告成！

python将图片按标签分入不同文件夹

以上这篇使用python将图片按标签分入不同文件夹的方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持移动技术网。

您可能感兴趣的文章:

如对本文有疑问，请在下面进行留言讨论，广大热心网友会与你互动！！ 点击进行留言回复

相关文章:

验证码：
最近更新的文章
大家感兴趣的文章
移动技术网