使用Python的lxml库修改XML文件时保留头部注释并覆盖原文件的实现方案

我懂你这个痛点——用lxml修改XML后，既要自动写回原文件，又得保住开头的注释和XML声明，手动对比太麻烦了。其实只要调整一下写入文件的方式，就能完美解决这个问题。

完整解决方案代码 #

from lxml import etree

# 1. 读取XML文件，完整保留所有节点（包括注释、XML声明）
tree = etree.parse("Example.xml")
root = tree.getroot()

# 2. 执行你的属性移除逻辑（示例：移除<Foo>的Pet属性）
if "Pet" in root.attrib:
    del root.attrib["Pet"]

# 3. 生成包含所有内容的XML字节流
xml_output = etree.tostring(
    tree,
    encoding='utf-8',
    xml_declaration=True,
    pretty_print=True
)

# 4. 写回原文件（用wb模式避免编码问题）
with open("Example.xml", "wb") as f:
    f.write(xml_output)

关键细节说明 #

• 读取阶段：etree.parse()会完整加载整个XML文档的结构，包括根元素前的注释、XML声明，不会像部分轻量解析库那样忽略这些非元素节点。
• 写入阶段：
  • xml_declaration=True：强制保留开头的<?xml version="1.0" encoding="utf-8"?>声明行
  • encoding='utf-8'：和原文件编码保持一致，避免乱码
  • pretty_print=True：保持原文件的换行、缩进格式（如果不需要严格保留属性间的空格，也可以去掉此参数，输出会变成紧凑格式）

这样处理后，修改后的XML会自动覆盖原文件，同时完美保留你需要的头部注释和声明，完全不需要手动对比操作。

内容的提问来源于stack exchange，提问作者Kevin Le