Python: avoid nested loop on array

Question

I am recursing through an xml file, using etree.

import xml.etree.ElementTree as etree
tree = etree.parse('x.xml')
root = tree.getroot()
for child in root[0]:
 for child in child.getchildren():
        for child in child.getchildren():
            for child in child.getchildren():
               print(child.attrib)

what is the idiomatic way in python to avoid these nested for loop.

  getchildren() ⇒ list of Element instances [#]
    Returns all subelements. The elements are returned in document order.

Returns:
A list of subelements.

I saw some post in SO like, Avoiding nested for loops but doesn't directly translate to my use.

thanks.

itertools.product is a nice way to avoid nested loops. Why doesn't that translate to your use? — David Cain
– David Cain, Commented Feb 8, 2013 at 20:13
Do you specifically want attributes for elements 4 children deep? — bogatron
– bogatron, Commented Feb 8, 2013 at 20:15
sorry, I didn't mean itertools.product doesn't suit me, but couldn't translate that example to arrays like in my case. I havn't done much Python, but will try. — bsr
– bsr, Commented Feb 8, 2013 at 20:30

Ord · Accepted Answer · 2013-02-08 20:38:40Z

3

If you want to get the children that are n levels deep in the tree, and then iterate through them, you can do:

def childrenAtLevel(tree, n):
    if n == 1:
        for child in tree.getchildren():
            yield child
    else:
        for child in tree.getchildren():
            for e in childrenAtLevel(child, n-1):
                yield e

Then, to get the elements four levels deep, you would simply say:

for e in childrenAtLevel(root, 4):
     # do something with e

Or, if you want to get all of the leaf nodes (i.e. the nodes that don't have any children themselves), you can do:

def getLeafNodes(tree):
    if len(tree) == 0:
         yield tree
    else:
         for child in tree.getchildren():
            for leaf in getLeafNodes(child):
                yield leaf

edited Feb 8, 2013 at 20:38

answered Feb 8, 2013 at 20:25

Ord

5,8535 gold badges31 silver badges44 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

Community · Accepted Answer · 2017-05-23 12:27:02Z

2

itertools.chain.from_iterable will flatten one level of nesting; you can use functools.reduce to apply it n times (Compressing "n"-time object member call):

from itertools import chain
from functools import reduce

for child in reduce(lambda x, _: chain.from_iterable(x), range(3), root):
    print(child.attrib)

Note that getchildren is deprecated; iterating a node yields its children directly.

edited May 23, 2017 at 12:27

CommunityBot

11 silver badge

answered Feb 8, 2013 at 20:40

ecatmur

158k28 gold badges311 silver badges387 bronze badges

Collectives™ on Stack Overflow

Python: avoid nested loop on array

2 Answers 2

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related