【Python】如何在文件夹里批量替换文本中的内容？

1.Introduction

用深度学习做目标检测时，有时候标签需要批量替换，这时一个批量处理的代码就很关键~

2.Materials and methods

下面对文件夹中的所有 xml 文件中的指定内容进行替换，txt 文件同理。

代码如下：

# -*- coding: utf-8 -*-
"""
Created on Fri Dec 25 19:07:33 2020

@author: YaoYee
"""
#coding=utf-8
import os

def listFiles(dirPath):
    fileList = []
    for root, dirs, files in os.walk(dirPath):
        for fileObj in files:
            fileList.append(os.path.join(root, fileObj))
    return fileList
    
def main():
    fileDir = "C:/Users/YaoYee/Desktop/val"
   #regex = ur'FUNC_SYS_ADD_ACCDETAIL'
    fileList = listFiles(fileDir)
    fileList = listFiles(fileDir)
    for fileObj in fileList:
        f = open(fileObj, 'r+')
        all_the_lines = f.readlines()
        f.seek(0)
        f.truncate()
        for line in all_the_lines:
            # e.g. 'dog' is original name, 'cat' is the new one.
            f.write(line.replace('dog', 'cat'))
        f.close()


if __name__ == '__main__':
    main()
    

3. Results and discussion

下面以把 dog 换成 cat 为例~

如果要同时替换多个内容文本呢？

用下面的代码替换下就ok了：

f.write(line.replace('dog', 'cat').replace('man', 'boy'))

4. Conclusion

Done is better than perfect.

猜你喜欢：👇🏻
⭐【Python】如何在文件夹里批量修改文件名（001-100）？
⭐【Python】如何在文件夹里批量分割图片？
⭐【Python】随机划分数据集并生成VOC格式列表