ElementTree库可用于创建和输出xml文件。首先创建根元素,再用SubElement添加子元素,通过text设置文本,attrib或set()添加属性;使用tostring输出字符串,ElementTree.write写入文件并指定编码和声明;结合minidom.prettify实现格式化缩进,使XML结构清晰易读。

python 中的 ElementTree 库是处理 XML 数据的标准工具之一,使用它生成 XML 文件简单高效。下面详细介绍如何用 ElementTree 创建和输出 XML 内容。
创建根元素并添加子元素
要生成 XML,先从创建根节点开始,然后逐级添加子元素。
- 导入 xml.etree.ElementTree 模块
- 使用 Element() 创建根节点
- 使用 SubElement() 添加子节点
示例代码:
from xml.etree.ElementTree import Element, SubElement, tostring
创建根元素
root = Element(‘bookstore’)
立即学习“Python免费学习笔记(深入)”;
添加第一个 book 子元素
book1 = SubElement(root, ‘book’, {‘category’: ‘fiction’})
title1 = SubElement(book1, ‘title’)
title1.text = ‘python编程入门’
author1 = SubElement(book1, ‘author’)
author1.text = ‘张三’
添加第二个 book
book2 = SubElement(root, ‘book’, {‘category’: ‘tech’})
title2 = SubElement(book2, ‘title’)
title2.text = ‘深入理解Python’
author2 = SubElement(book2, ‘author’)
author2.text = ‘李四’
输出生成的 XML 字符串
print(tostring(root, encoding=’unicode’))
将XML写入文件
生成的 XML 不仅可以打印到控制台,还能保存为 .xml 文件。
from xml.etree.ElementTree import Element, SubElement, ElementTree
root = Element(‘students’)
student1 = SubElement(root, ‘student’, {‘id’: ‘001’})
name1 = SubElement(student1, ‘name’)
name1.text = ‘王五’
age1 = SubElement(student1, ‘age’)
age1.text = ’20’
student2 = SubElement(root, ‘student’, {‘id’: ‘002’})
name2 = SubElement(student2, ‘name’)
name2.text = ‘赵六’
age2 = SubElement(student2, ‘age’)
age2.text = ’22’
使用 ElementTree 写入文件
tree = ElementTree(root)
tree.write(‘students.xml’, encoding=’utf-8′, xml_declaration=True)
执行后会生成 students.xml 文件,包含声明头和 UTF-8 编码。
设置文本内容与属性
Element 和 SubElement 支持直接设置标签文本和属性值。
- text 属性:用于设置标签内的文本内容
- attrib 参数:传入字典形式的属性(如 id、class 等)
- 也可通过 .set() 方法动态添加属性
例如:
book = Element(‘book’) book.set(‘published’, ‘2024’) book.text = ‘这是一本技术书籍’
格式化输出(美化XML)
默认生成的 XML 是紧凑的一行字符串。若需缩进格式便于阅读,可自定义美化函数。
import xml.etree.ElementTree as ET
def prettify(elem):
from xml.dom import minidom
rough_string = ET.tostring(elem, encoding=’unicode’)
reparsed = minidom.parseString(rough_string)
return reparsed.toprettyxml(indent=” “)
使用美化输出
print(prettify(root))
这样输出的 XML 会自动换行并带缩进,适合人工查看。
基本上就这些。ElementTree 足够轻量又功能完整,适合大多数 XML 生成需求。掌握创建、赋值、写入和美化四个步骤,就能灵活生成结构清晰的 XML 数据。不复杂但容易忽略细节,比如编码和声明的设置,建议每次写文件时都明确指定 encoding 和 xml_declaration。


