文章目录

• 前言
• 一、理论知识
• • 1.1 OS模块
  • 1.2 XML的解析
• 二、使用步骤
• • 1.引入库
  • 2.创建新文件夹
  • 3文件操作
  • 4 修改文件
• 总结

前言

本文要完成的是对于较多XML文档的自动修改，这部分往往在大量的图像标注的修改中会使用到，同时也不要局限于本文中所提到的使用场景。

一、理论知识

python中有大量的第三方库可以供我们使用，因此我们需要对所用到的库进行简单的学习，有大致印象即可，等到需要某个功能上网查一下就行了。
在python编程时，经常和文件、目录打交道，所以离不了os模块

1.1 OS模块

此部分网上有大量的教程，在此不再赘述。可以参考此文章：OS模块的学习

1.2 XML的解析

xml.etree.ElementTree模块实现了一个简单而高效的API用于解析和创建XML数据。 参考文章：XML

二、使用步骤

1.引入库

代码如下（示例）：

import xml.etree.ElementTree as ET
import os

2.创建新文件夹

代码如下（示例）：

xmldir = 'xml'
newxmldir = 'newxml'if not os.path.exists(newxmldir): #如果没有存放新xml的文件，那么
#就创建一个.os.makedirs(newxmldir)

3文件操作

for xml in os.listdir(xmldir):xmlpath = os.path.join(xmldir, xml)newxmlpath = os.path.join(newxmldir, xml)print("****************************")print("原始路径：   ",xmlpath)print("新文件的路径：", xmlpath)

上面代码实现的功能是遍历每一个文件，并找到原来的路径和存放新文件的路径。

得到这一步，下面要在上述的循环中进行修改操作

4 修改文件

tree = ET.parse(xmlpath)root = tree.getroot()  #root 有tag和attrib两个属性#print(root.tag,root.attrib)#for child in root:# print(child.tag, child.attrib)  查看树子支的标签for obj in root.findall('country'):name = obj.find('rank')if name.text == '1':name.text = '修改1'elif name.text == '4':name.text = '修改4'tree.write(newxmlpath, 'UTF-8')

上图是我们要修改的xml文档，假设一开始的文档中都是这种格式的数据。
观察可以发现其中的三个大框都是country，在这个节点的下面有rank,year,gdppc等子节点，所以假设我们要修改的为rank
修改结果如下：

具体的操作结合代码观看，一一对应的修改，不要盲目的找一个就用，因为数据格式还是多少有点区别的。

总结

本文实现的功能是批量化修改已有的xml文件。这种自动化的处理正是python的优雅之处，可以帮我们解决很多机械化的操作。
下面是全文的代码：（仅供参考）

#-*- coding: utf-8 -*-
import xml.etree.ElementTree as ET
import osxmldir = 'xml'
newxmldir = 'newxml'if not os.path.exists(newxmldir): #如果没有存放新xml的文件，那么就创建一个.os.makedirs(newxmldir)for xml in os.listdir(xmldir):xmlpath = os.path.join(xmldir, xml)newxmlpath = os.path.join(newxmldir, xml)print("****************************")print("原始路径：   ",xmlpath)print("新文件的路径：", xmlpath)tree = ET.parse(xmlpath)root = tree.getroot()  #root 有tag和attrib两个属性#print(root.tag,root.attrib)#for child in root:# print(child.tag, child.attrib)  查看树子支的标签for obj in root.findall('country'):name = obj.find('rank')if name.text == '1':name.text = '修改1'elif name.text == '4':name.text = '修改4'tree.write(newxmlpath, 'UTF-8')print(xmlpath + ' 已修改')

其中一部分print是用作测试功能的，可以省略的，只是为了更好的理解。