Python中如何利用to_xml为DataFrame生成的XML添加子元素或属性？

DevOps 小满

2026-8-22

解决方案

pandas的to_xml方法本身无法直接生成你需要的同级元素结构，因为它严格按照DataFrame的行列映射关系生成XML节点，默认会将单行数据转为指定的row_name节点（也就是你的<assesslet>），所有关联数据都会围绕这个节点展开。要实现需求，需要结合XML解析工具对生成的XML进行后续修改，具体步骤如下：

方法：生成XML后添加同级元素

先通过原有代码生成基础XML，再用Python内置的xml.etree.ElementTree库修改文件，添加目标元素：

示例代码

import xml.etree.ElementTree as ET
from pathlib import Path

# 1. 执行原有代码生成初始XML
(data.to_frame().T.to_xml(
    output_dir / output_file,
    attr_cols=["disabled", "error-if-not-time", "interval-type", "type", "verbose"],
    row_name="assesslet",
    pretty_print=True
))

# 2. 解析生成的XML文件
tree = ET.parse(output_dir / output_file)
root = tree.getroot()  # 获取根节点<data>

# 3. 创建新元素并设置属性（假设polarity值来自DataFrame的polarity列）
new_element = ET.Element("new_element")
new_element.set("polarity", str(data["polarity"].values[0]))  # 从DataFrame取对应值

# 4. 将新元素添加到<data>节点下
root.append(new_element)

# 5. 格式化并保存修改后的XML
ET.indent(tree, space="  ")  # Python 3.9+支持自动缩进格式化
tree.write(
    output_dir / output_file,
    encoding="utf-8",
    xml_declaration=True,
    method="xml"
)