遍历文件夹，将所需要的文本复制到人一个文件夹

王有理

# -*- coding:utf-8 -*-
import os
import os.path
import re
import  shutil
label_dir="D:\\工作任务\\wzg\\datasetTraffic\\Annotations"#已知文件
data_base_dir = open("D:\\工作任务\\wzg\\person1.txt","r")#给定的需要查找的txt
annotion_dir="D:\\工作任务\\wzg\\datasetTraffic\\ren"#得到的结果
lab = os.listdir(label_dir)
s = []
for i in data_base_dir:
    b = i.strip()
    for fileNum in lab:
        if not os.path.isdir(fileNum):
            a = fileNum
            if(b ==a):
                print(a)
                #shutil.copy(a ,annotion_dir)
                for file in os.listdir(label_dir):
                    if os.path.isfile(label_dir +'/'+file):
                        if b in file:                     
                            shutil.copy(label_dir +'/'+file, annotion_dir +'/'+file)