Python查询XML元素继承关系：代码输出空集合问题求助

核心问题分析 #

你的代码输出空集合，主要有以下几个关键问题：

1. 缩进语法错误：原函数内的tree = ET.parse(xml_file)和root = tree.getroot()未缩进，属于语法错误，若未修正会直接导致代码无法正常运行。
2. 元素匹配逻辑偏差：代码用elem.tag == element_name匹配目标元素，但结合继承关系的场景，你的XML大概率是用id属性标识元素名称，而非元素标签等于目标名称，这直接导致找不到目标元素，最终返回空集合。
3. 父元素查找无容错处理：当root.find(f".//*[@id='{parent}']")找不到对应元素时，直接调用.get('parent')会抛出AttributeError，中断遍历流程。

修复后的代码 #

假设你的XML结构是元素通过id属性定义名称，parent属性指向父元素的id（例如<node id="Child" parent="Parent"/>），修正后的代码如下：

import xml.etree.ElementTree as ET

def find_inheritance(element_name, xml_file):
    tree = ET.parse(xml_file)
    root = tree.getroot()

    elements = set()
    # 遍历所有元素，通过id属性匹配目标元素
    for elem in root.iter():
        if elem.get('id') == element_name:
            elements.add(element_name)
            parent_id = elem.get('parent')
            while parent_id is not None:
                elements.add(parent_id)
                # 查找父元素并增加容错判断
                parent_elem = root.find(f".//*[@id='{parent_id}']")
                if parent_elem is None:
                    break
                parent_id = parent_elem.get('parent', None)
            # 找到目标元素后退出遍历，提升效率
            break
    return elements

# 交互输入
element_name = input("Enter the name of the XML element: ")
xml_file = input("Enter the path of the XML file: ")

# 执行查询并输出结果
inheritance_elements = find_inheritance(element_name, xml_file)
print(f"Inheritance elements of {element_name}: {inheritance_elements}")

额外说明 #

• 如果你的XML中元素名称确实存储在tag上，只需把elem.get('id') == element_name改回elem.tag == element_name，同时确保输入的element_name与XML中的元素标签完全一致（注意大小写敏感）。
• 新增break语句，找到目标元素后立即退出遍历，避免处理无关元素，提升代码运行效率。
• 增加parent_elem非空判断，防止找不到父元素时抛出异常，保证遍历流程稳定。

内容的提问来源于stack exchange，提问作者Rahul cark