如何使用ElementTree以递归方式迭代Python中的XML标签?

时间:2014-01-12 12:01:15

标签: python xml

我正在尝试使用ElementTree迭代树中的所有节点。

我做的事情如下:

  tree = ET.parse("/tmp/test.xml")

  root = tree.getroot()

  for child in root:
       ### do something with child

问题是child是一个Element对象而不是ElementTree对象,所以我无法进一步查看它并递归迭代它的元素。有没有办法对“root”进行不同的迭代,以便迭代树中的顶级节点(直接子节点)并返回与root本身相同的类?

4 个答案:

答案 0 :(得分:24)

要迭代所有节点,请使用iter上的ElementTree方法,而不是根元素。

root是一个Element,就像树中的其他元素一样,只有它自己的属性和子元素的上下文。 ElementTree具有所有元素的上下文。

例如,给定此xml

<?xml version="1.0"?>
<data>
    <country name="Liechtenstein">
        <rank>1</rank>
        <year>2008</year>
        <gdppc>141100</gdppc>
        <neighbor name="Austria" direction="E"/>
        <neighbor name="Switzerland" direction="W"/>
    </country>
    <country name="Singapore">
        <rank>4</rank>
        <year>2011</year>
        <gdppc>59900</gdppc>
        <neighbor name="Malaysia" direction="N"/>
    </country>
    <country name="Panama">
        <rank>68</rank>
        <year>2011</year>
        <gdppc>13600</gdppc>
        <neighbor name="Costa Rica" direction="W"/>
        <neighbor name="Colombia" direction="E"/>
    </country>
</data>

您可以执行以下操作

>>> import xml.etree.ElementTree as ET
>>> tree = ET.parse('test.xml')
>>> for elem in tree.iter():
...     print elem
... 
<Element 'data' at 0x10b2d7b50>
<Element 'country' at 0x10b2d7b90>
<Element 'rank' at 0x10b2d7bd0>
<Element 'year' at 0x10b2d7c50>
<Element 'gdppc' at 0x10b2d7d10>
<Element 'neighbor' at 0x10b2d7e90>
<Element 'neighbor' at 0x10b2d7ed0>
<Element 'country' at 0x10b2d7f10>
<Element 'rank' at 0x10b2d7f50>
<Element 'year' at 0x10b2d7f90>
<Element 'gdppc' at 0x10b2d7fd0>
<Element 'neighbor' at 0x10b2db050>
<Element 'country' at 0x10b2db090>
<Element 'rank' at 0x10b2db0d0>
<Element 'year' at 0x10b2db110>
<Element 'gdppc' at 0x10b2db150>
<Element 'neighbor' at 0x10b2db190>
<Element 'neighbor' at 0x10b2db1d0>

答案 1 :(得分:6)

根据Robert Christie的回答,通过将Element转换为ElementTree,可以使用fromstring()遍历所有节点:

import xml.etree.ElementTree as ET

e = ET.ElementTree(ET.fromstring(xml_string))
for elt in e.iter():
    print "%s: '%s'" % (elt.tag, elt.text)

答案 2 :(得分:5)

您还可以访问以下特定元素:

country= tree.findall('.//country')

然后循环range(len(country)) 并访问

答案 3 :(得分:1)

除了罗伯特·克里斯蒂(Robert Christie)接受的答案之外,分别打印值和标签非常容易:

tree = ET.parse('test.xml')
for elem in tree.iter():
    print(elem.tag, elem.text)