Cómo recorrer un generador

¿Cómo se puede recorrer un generador? Pensé de esta manera:

gen = function_that_returns_a_generator(param1, param2)
if gen: # in case the generator is null
    while True:
        try:
            print gen.next()
        except StopIteration:
            break

¿Existe una forma más pitónica?

Simplemente

for x in gen:
    # whatever

hará el truco. Tenga en cuenta que if gensiempre vuelve True.

for item in function_that_returns_a_generator(param1, param2):
    print item

No necesita preocuparse por la prueba para ver si su función devuelve algo, ya que si no hay nada devuelto, no ingresará al ciclo.

En caso de que no necesite la salida del generador porque solo le importan sus efectos secundarios, puede usar el siguiente resumen:

for _ in gen: pass

Simplemente puede recorrerlo:

>>> gen = (i for i in range(1, 4))
>>> for i in gen: print i
1
2
3

Pero tenga en cuenta que solo puede realizar un bucle una vez. La próxima vez que el generador esté vacío:

>>> for i in gen: print i
>>> 

Simplemente trátelo como cualquier otro iterable:

for val in function_that_returns_a_generator(p1, p2):
    print val

Tenga en cuenta que if gen:siempre será Verdadero, por lo que es una prueba falsa

Si desea moverse manualmente a través del generador (es decir, trabajar con cada bucle manualmente), puede hacer algo como esto:

    from pdb import set_trace

    for x in gen:
        set_trace()
        #do whatever you want with x at the command prompt
        #use pdb commands to step through each loop of the generator e.g., >>c #continue   

Las otras respuestas son buenas para escenarios complicados. Si simplemente desea transmitir los elementos a una lista:

x = list(generator)

(o, si solo desea activar el generador para hacer cosas, simplemente list(generator).

Para un preprocesamiento simple, use listas por comprensión:

x = [tup[0] for tup in generator]

O cuando desee ejecutar funciones simples:

# didn't assign to variable b/c we don't care about what the print() function returns
[print(x) for x in gen]