使用 Python 的 ElementTree .iter() 时保持父子关系

Question

我有以下标记：

<a>
  <b>
    <c>
    <d>
    <e>
  </b>
  <f>
    <g>
  </f>
</a>

使用 ElementTree 函数 .iter() 我得到类似：

a, b, c, d, e, f, g

我需要找到一种方法来保持父母和孩子之间的关系，例如，我想知道“f”父母是“a”。到目前为止，我能想到的唯一方法是每次找到父节点时：len(list(elem)) > 0，我将该节点添加到列表中并跟踪当前的“级别" 的节点来建立这种关系。我觉得这个解决方案不是很优雅，我确信有一个更简单的解决方案，不幸的是我还没有找到它：/，我希望有人能对我有所启发：D

ps。在有人“在你问之前使用搜索”之前，我已经阅读了每一篇在某种程度上与我正在尝试做的事情相关的帖子，比如：

• Keep track of parent elements using ElementTree
• Iterating multiple (parent,child) nodes using Python ElementTree

碰巧他们是非常具体的用例，对我没有太大帮助，或者至少我没有找到将他们的解决方案与我的解决方案联系起来的方法。

提前致谢

Answer 1

您可以使用字典，它更适合树状结构。目标是让字典的键成为父级，值是子级列​​表。你可以这样做：

def get_children(parent):
    return [child for child in parent]

def get_parent_children_mapping(tree):
    return {parent: get_children(parent) for parent in tree.iter()}

示例用法如下：

import xml.etree.ElementTree as ET

def get_children(parent):
    return [child for child in parent]

def get_parent_children_mapping(tree):
    return { parent: get_children(parent) for parent in tree.iter() }

if __name__ == "__main__":

    s = """
    <a>
      <b>
        <c>Hello</c>
        <d>World</d>
        <e>Goodbye</e>
      </b>
      <f>
        <g>Hmmm...</g>
        <c>Hello</c>
      </f>
    </a>
    """

    tree = ET.fromstring(s)

    for parent, children in get_parent_children_mapping(tree).items():
        if children:
            print("{0} -> {1}".format(parent, children))

您会发现根元素被省略了——这是因为它显然没有父元素，但它的子元素是整个树上从get_parent_children_mapping 返回的所有元素。

看到它在行动here。只需确保您的 XML 有效。