Как рекурсивно перебирать теги XML в Python с помощью ElementTree?

Я пытаюсь выполнить итерацию по всем узлам в дереве с помощью ElementTree.

Я делаю что-то вроде:

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

  root = tree.getroot()

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

Проблема заключается в том, что дочерний элемент - это объект Element, а не объект ElementTree, поэтому я не могу вдаваться в него дальше и рекурсивно перебирать его элементы. Есть ли способ итерации по-разному над "root", чтобы он выполнял итерации над узлами верхнего уровня в дереве (непосредственные дочерние элементы) и возвращал тот же класс, что и сам root?

Ответ 1

Чтобы перебрать все узлы, используйте iter метод ElementTree, а не корневой элемент.

Корень - это Элемент, как и другие элементы в дереве, и только на самом деле имеет контекст его собственных атрибутов и детей. 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>

Ответ 2

вы также можете получить доступ к определенным элементам следующим образом:

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

затем перейдем к range(len(country)) и доступ

Ответ 3

Добавляя к ответу Роберту Кристи, можно перебирать все узлы с помощью fromstring() путем преобразования элемента в ElementTree:

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)

Ответ 4

В дополнение к принятому ответу Роберта Кристи распечатать значения и теги отдельно очень просто:

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