Discussion :

[olive_le_malin]

Bonjour,

j'utilise la librairie ElementTree pour parser un fichier XML, et je voudrais récupérer le nombre d'éléments total :

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

1
2
3
4
 
tree = parse(filename)
for node in tree.findall("toto")
...#peut-on connaître ici le nombre total d'éléments, sans avoir à incrémenter un compteur à chaque passage ?

merci par avance
@+

[dividee]

Je n'ai jamais utilisé ElementTree, mais je crois pouvoir te répondre uniquement sur base de la doc.

Dans la doc de la méthode findall de ElementTree:

Returns:
A list or iterator containing all matching elements, in section order.

Elle retourne donc une liste OU un itérateur.
Si c'est un liste qui est retournée, tu peux utiliser la fonction len pour avoir sa taille.
Si c'est un itérateur, non, il faut le parcourir.

Comme, d'après la doc, ce n'est pas forcément toujours une liste qui est retournée, tu ne peux pas l'assumer même si tu fais un test et que cela semble être le cas.

Donc, la réponse est non, il faut toujours prévoir le parcours, même si, pour optimiser, tu peux tester s'il s'agit d'une liste et dans ce cas appeler len:

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

1
2
3
4
5
6
7
totos = tree.findall('toto')
if isinstance(totos, list):
    nb_elems = len(totos)
else:
    nb_elems = 0
    for node in totos:
        nb_elems += 1

Attention de ne parcourir l'itérateur qu'une seule fois. Si tu utilises ce code tel quel, tu seras forcé d'appeler findall une seconde fois si tu souhaite faire un autre traitement sur le résultat... Mais dans ce cas, l'optimisation n'est pas utile vu que tu dois de toute façon parcourir l'itérateur ou la liste.

[DelphiManiac]

Même si tu as un itérateur, tu peux avoir la longueur, il suffit que tu cast le résultat par list(xx).

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

1
2
3
4
5
6
7
8
#!/usr/bin/env python
# -*- coding: utf-8 -*-
 
def test():
    for i in xrange(10):
        yield i
 
print len(list(test()))

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

1
2
3
4
5
6
7
#!/usr/bin/env python
# -*- coding: utf-8 -*-
 
def test():
    return [0, 1, 2, 3]
 
print len(list(test()))

Dans les 2 cas tu as la longueur, sans te préoccuper de savoir si c'est une liste ou non.

[olive_le_malin]

OK,
merci beaucoup à vous deux.
En fait, j'avoue que j'ai fait de nombreux essais, et j'obtiens tout le temps une liste, donc j'utilise len().

Merci
@+

[dividee]

merci, delphi, j'y avais pas pensé. Pour ma défense, la conversion (est-ce vraiment un cast ?) en liste parcourt tout de même l'itérateur

[DelphiManiac]

C'est pas un cast au sens C du terme, c'est une conversion et par contre cela duplique l'info.