如何在Python中过滤DataFrame并构建XML树？

1. 过滤DataFrame #

假设使用Pandas处理数据，先根据需求过滤掉不需要的行（示例为剔除空值行，可自行调整过滤条件）：

import pandas as pd
from lxml import etree

# 初始化示例DataFrame
df = pd.DataFrame({
    'DisplayName': ['firstname', 'lastname'],
    'Attribute': ['givenName', 'sn']
})

# 过滤DataFrame，保留DisplayName和Attribute均非空的行
filtered_df = df.dropna(subset=['DisplayName', 'Attribute'])

2. 动态构建XML结构 #

用lxml库生成XML，静态标签直接定义，遍历过滤后的DataFrame动态生成elmap标签：

# 创建XML根节点
root = etree.Element("root")

# 添加静态标签（示例结构，可替换为你的实际静态标签）
static_config = etree.SubElement(root, "config")
static_config.text = "固定配置内容"

static_mapping = etree.SubElement(root, "base_mapping")
static_mapping.text = "基础映射规则"

# 遍历过滤后的数据，生成elmap标签
for _, row in filtered_df.iterrows():
    elmap_node = etree.SubElement(root, "elmap")
    elmap_node.text = f"{row['DisplayName']}={row['Attribute']}"

# 生成格式化后的XML字符串
formatted_xml = etree.tostring(root, encoding='utf-8', pretty_print=True).decode('utf-8')
print(formatted_xml)

3. 输出结果示例 #

运行代码后会生成如下结构的XML：

<root>
  <config>固定配置内容</config>
  <base_mapping>基础映射规则</base_mapping>
  <elmap>firstname=givenName</elmap>
  <elmap>lastname=sn</elmap>
</root>

自定义说明 #

• 过滤逻辑可灵活调整，比如仅保留特定DisplayName值：filtered_df = df[df['DisplayName'].isin(['firstname', 'lastname'])]
• 若使用Python内置的xml.etree.ElementTree库，核心逻辑一致，仅创建节点的API略有差异

内容的提问来源于stack exchange，提问作者James Fowler