如何将平面xml数据转换为分层数据XML2

import xml.etree.ElementTree as ET

# 解析平面XML数据
tree = ET.parse('flat_data.xml')
root = tree.getroot()

# 创建分层数据结构
hierarchical_data = ET.Element('hierarchical_data')

# 遍历平面数据
for flat_node in root:
    # 提取数据
    data = flat_node.text
    
    # 创建分层节点
    hierarchical_node = ET.SubElement(hierarchical_data, 'node')
    hierarchical_node.text = data
    
    # 设置节点属性
    hierarchical_node.set('id', flat_node.get('id'))
    
    # 将节点插入到相应位置
    parent_id = flat_node.get('parent_id')
    if parent_id is not None:
        parent_node = hierarchical_data.find(".//*[@id='" + parent_id + "']")
        parent_node.append(hierarchical_node)
    else:
        hierarchical_data.append(hierarchical_node)

# 构建分层XML
hierarchical_tree = ET.ElementTree(hierarchical_data)

# 序列化分层XML
hierarchical_tree.write('hierarchical_data.xml', encoding='utf-8', xml_declaration=True)