Python利用ElementTree模块处理XML的方法详解

什么是ElementTree模块

ElementTree是Python标准库中用来解析和操作XML文件的模块，它是以树形结构来表示XML文档的层次结构。ElementTree具有简单易用、高效的特点，不仅可以解析XML文件，还可以创建、修改、删除XML元素、属性、文本等。

ElementTree模块的基本使用

导入ElementTree模块

要使用ElementTree模块，首先需要导入它。可以使用以下代码导入：

import xml.etree.ElementTree as ET

其中，xml.etree.ElementTree是ElementTree模块的完整名称，将它命名为ET可以方便地调用ElementTree模块中的方法。

创建ElementTree对象

要处理XML文件，首先需要将XML文件加载到内存中，然后使用ElementTree对象进行操作。可以使用以下代码创建ElementTree对象：

tree = ET.parse('my_xml_file.xml')

其中，my_xml_file.xml是文件名。这样就创建了一个名为tree的ElementTree对象。

获取根元素

ElementTree对象的根元素是所有XML元素的根，可以使用以下代码获取根元素：

root = tree.getroot()

其中，root是根元素的变量名。获取根元素后，就可以对根元素进行操作。根元素还有属性，可以使用以下代码获取根元素的属性：

root.attrib

遍历子元素

有两种方法可以遍历根元素的子元素。第一种方法是使用root元素的迭代器：

for child in root:
    print(child.tag, child.attrib)

其中，child.tag表示子元素的标签名，child.attrib表示子元素的属性。

第二种方法是使用root元素的findall()方法：

for child in root.findall('./child'):
    print(child.tag, child.attrib)

其中，'./child'表示查询根元素下所有名为child的子元素。

访问元素内容

使用element.text可以访问元素的文本内容。例如，以下是一个包含文本内容的XML片段<title>The title of the book</title>：

element = ET.Element('title')
element.text = 'The title of the book'

print(element.text)

输出结果为The title of the book。

修改元素内容

可以通过给元素的text属性重新赋值来修改元素的内容。例如：

element.text = '新的内容'

添加元素

可以使用ET.SubElement()方法向元素添加子元素。例如：

new_element = ET.SubElement(parent_element, 'child_element')

其中，parent_element是要添加子元素的元素，'child_element'是新元素的标签名。

删除元素

可以使用parent_element.remove(child_element)方法删除元素。其中，parent_element是子元素的父元素，child_element是要删除的元素。

示例

以下是一个包含学生信息的XML文件students.xml：

<?xml version="1.0" encoding="UTF-8"?>
<students>
  <student id="001">
    <name>张三</name>
    <gender>男</gender>
    <age>18</age>
  </student>
  <student id="002">
    <name>李四</name>
    <gender>女</gender>
    <age>19</age>
  </student>
  <student id="003">
    <name>王五</name>
    <gender>男</gender>
    <age>20</age>
  </student>
</students>

以下代码演示了如何使用ElementTree模块读取students.xml文件并输出学生信息：

import xml.etree.ElementTree as ET

# 创建ElementTree对象
tree = ET.parse('students.xml')

# 获取根元素
root = tree.getroot()

# 遍历子元素
for student in root.findall('student'):
    # 获取student元素的id属性
    id = student.get('id')

    # 获取name、gender、age子元素的值
    name = student.find('name').text
    gender = student.find('gender').text
    age = student.find('age').text

    # 输出学生信息
    print("学生ID：%s" % id)
    print("姓名：%s" % name)
    print("性别：%s" % gender)
    print("年龄：%s" % age)

输出结果为：

学生ID：001
姓名：张三
性别：男
年龄：18
学生ID：002
姓名：李四
性别：女
年龄：19
学生ID：003
姓名：王五
性别：男
年龄：20

以下代码演示了如何使用ElementTree模块向students.xml文件添加一个学生信息：

import xml.etree.ElementTree as ET

# 创建ElementTree对象
tree = ET.parse('students.xml')

# 获取根元素
root = tree.getroot()

# 创建新的student元素
new_student = ET.Element('student', attrib={'id': '004'})

# 创建name、gender、age子元素
name = ET.SubElement(new_student, 'name')
gender = ET.SubElement(new_student, 'gender')
age = ET.SubElement(new_student, 'age')

# 给子元素赋值
name.text = '赵六'
gender.text = '男'
age.text = '21'

# 将新元素添加到根元素中
root.append(new_student)

# 将修改后的树写入文件
tree.write('students.xml')

此代码执行后，students.xml文件将会增加一个学生信息。