已知需保留元素集合，如何从lxml树中移除其余元素？

我正在编写一个Python XML（NetConf）解析器，目标是从服务器获取rpc-reply XML，修改部分内容后生成可发送回服务器的最小配置XML。在GUI中修改值时，我会将修改的元素、它们的祖先元素以及无子节点的兄弟元素加入集合self.itemstokeep，这些是生成最小可行结果文件所需的内容。

处理的示例XML（简化版）如下：

<rpc-reply xmlns="urn:ietf:params:xml:ns:netconf:base:1.0" message-id="urn:uuid:a1cfef75-dba4-4fdf-81eb-8d5f65d35511">
  <data>
    <bridges xmlns="urn:ieee:std:802.1Q:yang:ieee802-dot1q-bridge">
      <bridge>
        (...)
      </bridge>
    </bridges>
    <interfaces xmlns="urn:ietf:params:xml:ns:yang:ietf-interfaces">
      <interface>
        <name>PORT_0</name>
        <description>random</description>
        <type xmlns:ianaift="urn:ietf:params:xml:ns:yang:iana-if-type">ianaift:ethernetCsmacd</type>
        <bridge-port xmlns="urn:ieee:std:802.1Q:yang:ieee802-dot1q-bridge">
            (...)
        </bridge-port>
      </interface>
      <interface>
        (...)
      </interface>
    </interfaces>
    <keystore xmlns="urn:ietf:params:xml:ns:yang:ietf-keystore">
        (...)
    </keystore>
  </data>
</rpc-reply>

我在结合使用.iter()和.remove()时遇到问题：例如修改<description>时，仅会移除<bridge>分支，之后.iter()循环无法回到<interfaces>或其直接祖先，大概率是因为移除元素后丢失了祖先信息，即循环在第一个遇到的“最后叶子”元素处停止。

我当前使用如下代码移除元素，self.itemstokeep是需保留的etree.Element集合：

for item in treecopy.iter():
    if not item in self.itemstokeep:
        if not item.getparent() == None:
            item.getparent().remove(item)
    else:
        continue

能否推荐合适的解决方法或替代方案？与现有方案不同的是，我只知道需保留的元素，不清楚要移除哪些，且除两个顶层元素外输入结构不固定，常规xpath方法较复杂。我也曾考虑过放弃itemstokeep集合，在修改时重建树，但担心需频繁检查祖先重复、多次遍历树导致性能不佳，或许我忽略了某些优化点？

原代码的问题在于：.iter()是前序遍历，遍历过程中删除元素会破坏迭代器的结构，导致后续元素无法被正常访问。以下是几种可行的解决思路：

1. 先收集待删除元素，再批量移除 #

先遍历所有元素，把不在self.itemstokeep且有父节点的元素统一收集，之后再批量删除，避免遍历过程中修改树结构：

to_remove = []
for item in treecopy.iter():
    if item not in self.itemstokeep and item.getparent() is not None:
        to_remove.append(item)

for item in to_remove:
    item.getparent().remove(item)

这种方法能确保所有元素都被检查到，不会出现遍历中断的问题。

2. 采用后序遍历删除元素 #

使用后序遍历（先处理子节点，再处理父节点），这样即使删除子节点，父节点的遍历也不会受影响：

def post_order_iter(element):
    for child in list(element):
        yield from post_order_iter(child)
    yield element

for item in post_order_iter(treecopy.getroot()):
    if item not in self.itemstokeep and item.getparent() is not None:
        item.getparent().remove(item)

后序遍历能保证在处理父节点前，所有子节点都已完成检查和处理，适配结构不固定的XML场景。

3. 基于保留集合优化重建树 #

如果你选择重建树，可以通过标记法避免重复检查祖先，降低性能开销：

1. 给所有需保留的元素添加临时标记：

for item in self.itemstokeep:
    item.attrib['__keep__'] = '1'

2. 递归修剪树，保留带标记的元素，同时清理临时标记：

def prune_tree(element):
    # 先处理子节点
    children_to_keep = []
    for child in list(element):
        prune_tree(child)
        if '__keep__' in child.attrib:
            del child.attrib['__keep__']
            children_to_keep.append(child)
        else:
            element.remove(child)
    # 处理当前节点：若有标记则保留，无标记且无子节点则删除
    if '__keep__' in element.attrib:
        del element.attrib['__keep__']
    elif not element.getchildren():
        if element.getparent() is not None:
            element.getparent().remove(element)

prune_tree(treecopy.getroot())

这种方式仅需一次递归遍历，逻辑清晰且性能可控。

额外注意点 #

• 处理NetConf XML时，要注意命名空间的准确性，etree.Element的比较会同时考虑命名空间和标签名。
• 若self.itemstokeep包含父元素，其未被标记的子节点会被自动删除，符合“最小配置”的需求。

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