Parsing xml with lxml in python 3

Question

I have this code, i want to group animals with the same tags to the one group ex. tags<dog><dog> to <dogs><dog/><dog/></dogs> etc. But in my code, I have no idea why is the output without animals.

OUTPUT:

<root>
       <zoo>
           <some_tag/><some_diff/>
       </zoo>
       <zoo>
           <b/><o/>
       </zoo>
 </root>

CODE:

 xml = '`<root>
                  <zoo>
                      <some_tag/><some_diff/>
                      <dog/><dog/>
                      <cat/><cat/><cat/>
                  </zoo>
                  <zoo>
                      <b/><o/>
                      <dog/><dog/>
                      <cat/><cat/><cat/><cat/>
                  </zoo>
            </root>`'

from lxml import etree as et
root = et.fromstring(xml)
node = root.findall('./zoo')
j = False
k = False
for zoo in node:
    for animal in zoo:
        if 'dog' in animal.tag:
            if not j:
                dogs = et.SubElement(zoo,'dogs')
            dogs.append(animal)
            j = True
        if 'cat' in animal.tag:                        
            if not k:
                cats = et.SubElement(zoo,'cats')            
            cats.append(animal)
            k = True

    k = False
    j= False

rephrase your question please .. it is not so clear

securecurve
– securecurve

2013-10-08 23:36:25 +00:00
Commented Oct 8, 2013 at 23:36 — securecurve
– securecurve, Commented Oct 8, 2013 at 23:36
Ya .. much better :))

securecurve
– securecurve

2013-10-09 00:22:58 +00:00
Commented Oct 9, 2013 at 0:22 — securecurve
– securecurve, Commented Oct 9, 2013 at 0:22

securecurve · Accepted Answer · 2013-10-09 02:40:45Z

1

I made some modifications to your script, it works for me .. check it out:

xml = '''<root>
                  <zoo>
                      <some_tag/>
                      <some_diff/>
                      <dog/>
                      <dog/>
                      <cat/>
                      <cat/>
                      <cat/>
                  </zoo>

                  <zoo>
                      <b/>
                      <o/>
                      <dog/>
                      <dog/>
                      <cat></cat>
                      <cat></cat>
                  </zoo>
            </root>'''

from lxml import etree as et


root = et.fromstring(xml)

# The below 3 lines have the same effect, use whichever you like
node = root.findall('./zoo')
node = list( root.getchildren() )
node = root.getchildren()


dogs_flag = False
cats_flag = False

for zoo in node:

    # Resetting the flags in each iteration, otherwise, you will 
    # have all the cats and dogs inside one zoo element ... try it yourself
    dogs_flag = False
    cats_flag = False

    for animal in zoo:

        if 'dog' == animal.tag:
            if not dogs_flag:
                dogs = et.SubElement(zoo,'dogs')
                dogs_flag = True    # I think this is a better place to set your flag                

            dogs.append(animal)


        if 'cat' == animal.tag:                        
            if not cats_flag:
                cats = et.SubElement(zoo,'cats')            
                cats_flag = True

            cats.append(animal)


print et.tostring(root, pretty_print = True)

This will give you this output

        <root>
              <zoo>
                  <some_tag/>
                  <some_diff/>
                  <dogs>
                      <dog/>
                      <dog/>
                  </dogs>
                  <cats>
                      <cat/>
                      <cat/>
                      <cat/>
                  </cats>
              </zoo>

              <zoo>
                  <b/>
                  <o/>
                  <dogs>
                      <dog/>
                      <dog/>
                  </dogs>
                  <cats>
                      <cat/>
                      <cat/>
                  </cats>
              </zoo>
        </root>

edited Oct 9, 2013 at 2:40

answered Oct 9, 2013 at 1:26

securecurve

5,8276 gold badges49 silver badges84 bronze badges

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

2 Comments

bagere Over a year ago

Yep, that`s. Your code looks much better than mine. But my problem was with comparision. My code: 'dog' in animal.tag and in yours: 'dog' == animal.tag, do you know a reason why was my comparision bad?

securecurve Over a year ago

@dusan, usually in is used to find some object (everything in python is an object) in a string/list/tuple/set, but not for comparison between objects .. doing some experiments using in will help you to understand the difference... Good luck.

Collectives™ on Stack Overflow

Parsing xml with lxml in python 3

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related