在Python的标准库中,ElementTree是处理XML数据的轻量高效工具,无需额外安装依赖即可使用,适合各类XML解析场景。当我们需要从XML中筛选符合特定条件的节点,提取其属性值并整理成列表时,ElementTree的节点遍历和属性访问能力可以很好地满足需求。

ElementTree基础使用流程
使用ElementTree解析XML首先需要导入模块,然后加载XML数据,可以是本地文件也可以是字符串格式的内容。解析完成后会得到一个根节点对象,后续所有节点操作都基于这个根节点展开。
加载XML数据
如果是解析本地XML文件,使用parse方法;如果是解析XML字符串,使用fromstring方法,两种方式得到的根节点操作逻辑完全一致。
import xml.etree.ElementTree as ET
# 解析XML字符串示例
xml_str = """<root>
<user id="1" role="admin" name="张三"/>
<user id="2" role="user" name="李四"/>
<user id="3" role="admin" name="王五"/>
<user id="4" role="user" name="赵六"/>
</root>"""
root = ET.fromstring(xml_str)
# 解析本地文件示例,假设文件名为data.xml
# tree = ET.parse("data.xml")
# root = tree.getroot()
条件式提取属性并构建列表
提取属性的核心是通过节点对象的get方法获取指定属性值,条件筛选则可以通过遍历节点时判断属性值是否符合要求实现,最后将符合条件的属性值添加到列表中即可。
遍历节点与条件判断
ElementTree提供了findall方法可以快速查找所有符合标签名的子节点,也可以直接遍历根节点的子节点。我们可以在遍历过程中判断节点的属性是否满足筛选条件,比如筛选role为admin的用户id。
import xml.etree.ElementTree as ET
xml_str = """<root>
<user id="1" role="admin" name="张三"/>
<user id="2" role="user" name="李四"/>
<user id="3" role="admin" name="王五"/>
<user id="4" role="user" name="赵六"/>
</root>"""
root = ET.fromstring(xml_str)
# 初始化结果列表
admin_id_list = []
# 遍历所有user节点
for user_node in root.findall("user"):
# 获取role属性值
user_role = user_node.get("role")
# 条件判断:role为admin时提取id属性
if user_role == "admin":
user_id = user_node.get("id")
admin_id_list.append(user_id)
print(admin_id_list) # 输出结果:['1', '3']
提取多个属性构建复杂列表
如果需要提取多个属性,比如同时获取admin角色的用户的id和name,可以构建字典列表或者元组列表,满足更复杂的业务需求。
import xml.etree.ElementTree as ET
xml_str = """<root>
<user id="1" role="admin" name="张三"/>
<user id="2" role="user" name="李四"/>
<user id="3" role="admin" name="王五"/>
<user id="4" role="user" name="赵六"/>
</root>"""
root = ET.fromstring(xml_str)
# 构建字典列表存储多个属性
admin_info_list = []
for user_node in root.findall("user"):
if user_node.get("role") == "admin":
user_info = {
"id": user_node.get("id"),
"name": user_node.get("name")
}
admin_info_list.append(user_info)
print(admin_info_list) # 输出结果:[{'id': '1', 'name': '张三'}, {'id': '3', 'name': '王五'}]
注意事项
- 使用
get方法获取属性时,如果属性不存在会返回None,不会抛出异常,需要提前做好空值处理避免后续逻辑出错。 - 如果XML结构层级较深,可以使用
find方法逐层查找节点,或者使用XPath语法(ElementTree支持部分XPath语法)快速定位深层节点。 - 解析超大XML文件时,建议使用
iterparse方法实现增量解析,避免一次性加载全部内容占用过多内存。
ElementTree的get方法第二个参数可以设置默认值,比如user_node.get("role", "default"),当role属性不存在时返回默认值,减少空值判断的代码量。
ElementTreeXML解析属性提取列表构建Python修改时间:2026-07-13 05:54:22