如何使用Python编辑XML中的特定COUNTRY区块？

问题背景 #

原始XML内容：

<WESTERN>
            <TYPE>ABC</TYPE>
            <TYPE>MNO</TYPE>
            <COUNTRY>
                <NAME>MONACO</NAME>
                <DETAILS>            
                     <EUROPE CAPITAL="Monaco" />
                     <EUROPE population= "123456" />
                </DETAILS> 
            </COUNTRY>
            <COUNTRY>
                <NAME>MALTA</NAME>
                <DETAILS>
                    <EUROPE CAPITAL="Valletta" />
                    <EUROPE population= "123456" />
                </DETAILS>
                <DETAILS>
                    <EUROPE CONTINENT="EUROPE" />
                    <EUROPE GDP= "66666666"  />
                </DETAILS>
                <DETAILS>
                    <EUROPE CLIMATE="Warm" />
                    <EUROPE Votes= "123" />
                </DETAILS>
            </COUNTRY>
            <COUNTRY>
                 <NAME>ANDORRA</NAME>
                 <DETAILS>
                   <EUROPE CAPITAL="Andorra la Vella" />
                   <EUROPE population= "123456" />
                 </DETAILS>
           </COUNTRY>
        </WESTERN>

需求：仅在中间的<COUNTRY>（即第二个，索引为1）及其下所有<DETAILS>标签内添加<EUROPE tag="NEW"/>标签，预期效果如下：

<WESTERN>
            <TYPE>ABC</TYPE>
            <TYPE>MNO</TYPE>
            <COUNTRY>
                <NAME>MONACO</NAME>
                <DETAILS>            
                     <EUROPE CAPITAL="Monaco" />
                     <EUROPE population= "123456" />
                </DETAILS> 
            </COUNTRY>
            <COUNTRY>
                <NAME>MALTA</NAME>
                <DETAILS>
                    <EUROPE CAPITAL="Valletta" />
                    <EUROPE population= "123456" />

                    <EUROPE tag = "NEW"/>
                </DETAILS>
                <DETAILS>
                    <EUROPE CONTINENT="EUROPE" />
                    <EUROPE GDP= "66666666" />

                    <EUROPE tag = "NEW"/>
                </DETAILS>
                <DETAILS>
                    <EUROPE CLIMATE="Warm" />
                    <EUROPE Votes= "123" />

                    <EUROPE tag = "NEW"/>
                </DETAILS>
            </COUNTRY>
            <COUNTRY>
                 <NAME>ANDORRA</NAME>
                 <DETAILS>
                   <EUROPE CAPITAL="Andorra la Vella" />
                   <EUROPE population= "123456" />
                 </DETAILS>
           </COUNTRY>
        </WESTERN>

尝试的代码（问题：新标签被添加到所有<COUNTRY>/<DETAILS>中）：

tree = ET.parse('abc.xml')
root = tree.getroot()
tasks = tree.findall(".//COUNTRY")[1:2]
new_entry = tree.findall(".//DETAILS")
for i in new_entry:
    i.append(ET.fromstring('<EUROPE tag = "NEW"/>'))

解决方案 #

你的代码问题在于：tree.findall(".//DETAILS")会匹配XML中所有的<DETAILS>标签，没有限定范围到目标<COUNTRY>。正确的做法是先定位到目标<COUNTRY>，再在这个节点内部查找<DETAILS>标签：

import xml.etree.ElementTree as ET

tree = ET.parse('abc.xml')
root = tree.getroot()

# 定位到第二个COUNTRY节点（索引为1）
target_country = root.findall(".//COUNTRY")[1]

# 在目标COUNTRY内部查找所有DETAILS标签
details_list = target_country.findall(".//DETAILS")

# 为每个DETAILS添加新标签
for detail in details_list:
    # 每次创建新的标签对象，避免同一元素被重复添加
    detail.append(ET.fromstring('<EUROPE tag = "NEW"/>'))

# 保存修改后的XML
tree.write('updated_abc.xml', encoding='utf-8', xml_declaration=True)

关键逻辑说明 #

• 精准定位目标节点：通过root.findall(".//COUNTRY")[1]获取第二个<COUNTRY>节点，限定后续操作的范围。
• 范围查找子节点：调用目标<COUNTRY>的findall方法，仅查找该节点下的<DETAILS>，而非整个XML文档。
• 避免元素重复引用：每次添加时重新创建<EUROPE tag="NEW"/>对象，防止同一元素被多次追加（ElementTree中一个元素只能属于一个父节点）。

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