python 脚本 按照标签分类

将文件夹中的所有图片，按照标签文件夹内的分类，将所有图片按照类别划分到不同的文件夹
思路流程图
1.读取标签文件夹中的所有文件
2.

import os
import sys
import shutil#用于移动文件
import pdb
path="label_tube"
filenames=os.listdir(path)#读取标签文件


i=0
#pdb.set_trace()
for filename in filenames:
    with open("label_tube/"+filename,"r") as f:   
       line= f.readline() #读取每个文件的前6个标签字符
       label=line[:5]
       # 去除文件名字后的后缀
       index=filename.rfind('.')
       filename=filename[:index]
       if label=="hzshs" :
        #读取每张图片将图片存入img_tube_mm/hunzhuo
          shutil.move('img_tube/'+filename+'.jpg','img_tube_mm/hunzhuo')
         # shutil.move('label_tube/'+filename+'.txt','img_tube_mm/label')

       elif label=="qcshs":
        #将图片存入img_tube_mm/qingche
          shutil.move('img_tube/'+filename+'.jpg','img_tube_mm/qingche')
          #shutil.move('label_tube/'+filename+'.txt','img_tube_mm/label')