使用python批量处理文本文件中指定的内容

使用python批量处理文本文件中指定的内容

• 代码背景
  给数据集打标签时，搭上了错误的标签，不利于后续进行图像的识别。因此需要在已经生成的xml文件中修改标签的名称，以便于后续模型的训练。
• 代码

import os
path = "E:\ship" #文件夹目录
files= os.listdir(path) #得到文件夹下的所有文件名称
s = []
i=0
for file in files: #遍历文件夹
     if not os.path.isdir(file): #判断是否是文件夹，不是文件夹才打开
          f = open(path+"/"+file); #打开文件
          iter_f = iter(f); #创建迭代器
          str = ""
          for line in iter_f: #遍历文件，一行行遍历，读取文本
              if '		<name>ferry_boat</name>' in line:
                  line = line.replace('		<name>ferry_boat</name>','		<name>boat</name>')
              str = str + line
          s.append(str) #每个文件的文本存到list中
#print(s[1])

new_list = []
for file in files: #遍历文件夹
     if not os.path.isdir(file): #判断是否是文件夹，不是文件夹才打开
        with open(path+"/"+file,'w') as f:#覆盖写        
            f.write(s[i])
            i+=1