python - with - Bucle sobre dos generador juntos

yield from python

Question

Tengo dos generadores que dicen A() y B() . Quiero recorrer ambos generadores juntos. Algo como:

for a,b in A(),B():    # I know this is wrong
    #do processing on a and b

Una forma es almacenar los resultados de ambas funciones en las listas y luego recorrer la lista fusionada. Algo como esto:

resA = [a for a in A()]
resB = [b for b in B()]
for a,b in zip(resA, resB):
    #do stuff

Si se está preguntando, entonces sí, ambas funciones producen igual número de valor.

Pero no puedo usar este enfoque porque A()/B() devuelve tantos valores. Almacenarlos en una lista agotaría la memoria, por eso uso generadores.

¿Hay alguna manera de pasar por encima de ambos generadores a la vez?

Answer 1

Estabas casi allí. En Python 3, simplemente pase los generadores a zip() :

for a, b in zip(A(), B()):

zip() toma cualquier iterable, no solo listas. Se consumirán los generadores uno por uno.

En Python 2, usa itertools.izip() :

from itertools import izip

for a, b in izip(A(), B()):

Además, convertir un generador en una lista es tan simple como una list(generator) ; No hay necesidad de usar una lista de comprensión allí.

Answer 2

Puede usar la función zip para transformar las dos listas generadas en una lista de tuplas, de modo que puede usar la forma que desee:

for a,b in zip(A(), B()):
    pass

Answer 3

Suena como lo quieres itertools.izip() :

from itertools import izip

for a, b in izip(A(), B()):

De los documentos:

Como zip() excepto que devuelve un iterador en lugar de una lista.

De esta manera, nunca creas una list , ya sea de A() , B() o la izip() .

Tenga en cuenta que en el zip básico de Python 3 es como itertools.izip Python 2.x.