Python:LXML如何向现有元素树添加子元素以修改XML文件

没问题，我来帮你搞定用lxml给已有XML元素添加子元素的需求。下面我会结合你提供的XML内容，给出具体的实现步骤和代码示例，你可以直接套用或者根据实际需求调整。

原XML内容 #

<addressbook>
  <person>
    <name>Eric Idle</name>
    <phone type='fix'>999-999-999</phone>
    <phone type='mobile'>555-555-555</phone>
    <address>
      <street>12, spam road</street>
      <city>London</city>
      <zip>H4B 1X3</zip>
    </address>
  </person>
</addressbook>

实现步骤与代码示例 #

首先确保你已经安装了lxml库，如果没有的话先执行安装命令：

pip install lxml

接下来用Python代码完成修改操作，我会演示两种常见的添加场景：给<person>加<email>子元素，给<address>加<country>子元素：

from lxml import etree

# 解析XML内容（如果是从本地文件读取，替换成 etree.parse("你的文件名.xml").getroot()）
xml_str = """
<addressbook>
  <person>
    <name>Eric Idle</name>
    <phone type='fix'>999-999-999</phone>
    <phone type='mobile'>555-555-555</phone>
    <address>
      <street>12, spam road</street>
      <city>London</city>
      <zip>H4B 1X3</zip>
    </address>
  </person>
</addressbook>
"""
root = etree.fromstring(xml_str)

# 场景1：给第一个<person>元素添加<email>子元素
target_person = root.find("person")
# 创建新元素并设置内容
email_elem = etree.Element("email")
email_elem.text = "eric.idle@montypython.com"
# 还可以给元素加属性，比如：email_elem.set("type", "work")
# 将新元素添加到目标元素末尾
target_person.append(email_elem)

# 场景2：给<address>元素添加<country>子元素
target_address = root.find(".//address")  # 用xpath语法定位更灵活
country_elem = etree.Element("country")
country_elem.text = "United Kingdom"
target_address.append(country_elem)

# 输出修改后的XML（如果要保存到文件，用下面的write方法）
# etree.ElementTree(root).write("修改后的文件.xml", encoding='utf-8', xml_declaration=True, pretty_print=True)
print(etree.tostring(root, encoding='utf-8', pretty_print=True).decode('utf-8'))

代码关键点解释 #

• XML解析：如果你的XML是本地文件，直接用etree.parse("文件路径").getroot()获取根节点即可；如果是字符串内容，用etree.fromstring()解析。
• 元素定位：find()方法用于查找第一个匹配的元素；如果要找所有符合条件的元素，用xpath()，比如root.xpath("//person")会返回所有<person>元素的列表。
• 元素创建与添加：用etree.Element("元素名")创建新元素，text属性设置元素文本内容，set()方法添加属性；append()把新元素加到目标元素的子节点末尾，要是想插入到指定位置，用insert(索引位置, 元素)。
• 保存/输出：etree.tostring()用于输出XML字符串，write()方法可以直接保存到文件，pretty_print=True能让输出的XML格式更整洁。

运行代码后，你会得到修改后的XML：

<addressbook>
  <person>
    <name>Eric Idle</name>
    <phone type="fix">999-999-999</phone>
    <phone type="mobile">555-555-555</phone>
    <address>
      <street>12, spam road</street>
      <city>London</city>
      <zip>H4B 1X3</zip>
      <country>United Kingdom</country>
    </address>
    <email>eric.idle@montypython.com</email>
  </person>
</addressbook>

你可以根据自己的需求，修改要添加的元素名称、内容，或者调整定位目标元素的方式~

内容的提问来源于stack exchange，提问作者Greg Symsym