Comment parcourir deux listes en parallèle?

J'ai deux itérables en Python, et je veux les parcourir par paires:

foo = (1, 2, 3)
bar = (4, 5, 6)

for (f, b) in some_iterator(foo, bar):
    print "f: ", f, "; b: ", b

Il devrait en résulter:

f: 1; b: 4
f: 2; b: 5
f: 3; b: 6

Une façon de le faire est d'itérer sur les indices:

for i in xrange(len(foo)):
    print "f: ", foo[i], "; b: ", b[i]

Mais cela me semble quelque peu impythonique. Y a-t-il une meilleure façon de le faire?

Python 3

for f, b in zip(foo, bar):
    print(f, b)

zips'arrête lorsque le plus court de fooou bars'arrête.

En Python 3 , zip retourne un itérateur de tuples, comme itertools.izipen Python2. Pour obtenir une liste de tuples, utilisez list(zip(foo, bar)). Et pour compresser jusqu'à épuisement des deux itérateurs, vous utiliserez itertools.zip_longest .

Python 2

En Python 2 , zip retourne une liste de tuples. C'est bien quand fooet ce barn'est pas massif. S'ils sont tous les deux massifs, la formation zip(foo,bar)est une variable temporaire inutilement massive et doit être remplacée par itertools.izipou itertools.izip_longest, qui renvoie un itérateur au lieu d'une liste.

import itertools
for f,b in itertools.izip(foo,bar):
    print(f,b)
for f,b in itertools.izip_longest(foo,bar):
    print(f,b)

izips'arrête lorsque l'un fooou l' autre barest épuisé. izip_longests'arrête lorsque les deux fooet barsont épuisés. Lorsque le ou les itérateurs les plus courts sont épuisés, izip_longestdonne un tuple avec Nonela position correspondant à cet itérateur. Vous pouvez également définir un autre en fillvalueplus Nonesi vous le souhaitez. Voir ici pour l' histoire complète .

Notez également que zipet ses zipfrères semblables peuvent accepter un nombre arbitraire d'itérables comme arguments. Par exemple,

for num, cheese, color in zip([1,2,3], ['manchego', 'stilton', 'brie'], 
                              ['red', 'blue', 'green']):
    print('{} {} {}'.format(num, color, cheese))

impressions

1 red manchego
2 blue stilton
3 green brie

Vous voulez la zipfonction.

for (f,b) in zip(foo, bar):
    print "f: ", f ,"; b: ", b

Vous devez utiliser la fonction ' zip '. Voici un exemple à quoi peut ressembler votre propre fonction zip

def custom_zip(seq1, seq2):
    it1 = iter(seq1)
    it2 = iter(seq2)
    while True:
        yield next(it1), next(it2)

Vous pouvez regrouper les nièmes éléments dans un tuple ou une liste en utilisant la compréhension, puis les distribuer avec une fonction de générateur.

def iterate_multi(*lists):
    for i in range(min(map(len,lists))):
        yield tuple(l[i] for l in lists)

for l1, l2, l3 in iterate_multi([1,2,3],[4,5,6],[7,8,9]):
    print(str(l1)+","+str(l2)+","+str(l3))

Si quelqu'un cherche quelque chose comme ça, je l'ai trouvé très simple et facile:

list_1 = ["Hello", "World"]
list_2 = [1, 2, 3]

for a,b in [(list_1, list_2)]:
    for element_a in a:
        ...
    for element_b in b:
        ...

>> Hello
World
1
2
3

Les listes seront itérées avec leur contenu complet, contrairement à zip () qui n'itère que jusqu'à la longueur minimale du contenu.

Voici comment procéder avec la compréhension de la liste:

a = (1, 2, 3)
b = (4, 5, 6)
[print('f:', i, '; b', j) for i, j in zip(a, b)]

impressions:

f: 1 ; b 4
f: 2 ; b 5
f: 3 ; b 6