enumerate () - ing un generador en Python

88

Me gustaría saber qué sucede cuando paso el resultado de una función generadora a enumerate () de python. Ejemplo:

def veryBigHello():
    i = 0
    while i < 10000000:
        i += 1
        yield "hello"

numbered = enumerate(veryBigHello())
for i, word in numbered:
    print i, word

¿Se repite la enumeración de forma perezosa, o lo engulle todo en la primera? Estoy 99,999% seguro de que es perezoso, así que ¿puedo tratarlo exactamente igual que la función del generador, o debo tener cuidado con algo?

Adán
fuente
1
Supongo que quieres incrementar i en veryBigHello.
robert
@robert: si no me equivoco, se aumenta automáticamente
the_drow
@the_drow No en la veryBigHellofunción en sí.
Will McCutchen
1
@Will: Oh, correcto. Pero eso es simplemente quisquilloso. Es un ejemplo. Arreglado de todos modos.
the_drow

Respuestas:

103

Es perezoso. Es bastante fácil demostrar que ese es el caso:

>>> def abc():
...     letters = ['a','b','c']
...     for letter in letters:
...         print letter
...         yield letter
...
>>> numbered = enumerate(abc())
>>> for i, word in numbered:
...     print i, word
...
a
0 a
b
1 b
c
2 c
Dave Webb
fuente
¿Es este Python 2 o 3 (o ambos)? ¿Es perezoso en ambos? Probé en Python 2 y es vago.
becko
2
Probé esto en Python 3.5.2 y se evalúa de manera perezosa.
gobernador
42

Es incluso más fácil de decir que cualquiera de las sugerencias anteriores:

$ python
Python 2.5.5 (r255:77872, Mar 15 2010, 00:43:13)
[GCC 4.3.4 20090804 (release) 1] on cygwin
Type "help", "copyright", "credits" or "license" for more information.
>>> abc = (letter for letter in 'abc')
>>> abc
<generator object at 0x7ff29d8c>
>>> numbered = enumerate(abc)
>>> numbered
<enumerate object at 0x7ff29e2c>

Si enumerate no realiza una evaluación perezosa, devolvería [(0,'a'), (1,'b'), (2,'c')]o algo (casi) equivalente.

Por supuesto, enumerate es realmente un generador elegante:

def myenumerate(iterable):
   count = 0
   for _ in iterable:
      yield (count, _)
      count += 1

for i, val in myenumerate((letter for letter in 'abc')):
    print i, val
Wayne Werner
fuente
2
Gracias por esta explicacion. Me costó un poco averiguar la respuesta aceptada. Al menos hasta que vi el tuyo.
trendsetter37
13

Como puede llamar a esta función sin salirse de las excepciones de memoria, definitivamente es perezoso

def veryBigHello():
    i = 0
    while i < 1000000000000000000000000000:
        yield "hello"

numbered = enumerate(veryBigHello())
for i, word in numbered:
    print i, word
Nikolaus Gradwohl
fuente
0

Alternativa de la vieja escuela ya que estaba usando un generador que alguien más (sklearn) escribió que no funcionó con los enfoques aquí.

i=(-1)
for x in some_generator:
    i+=1
HashRocketSintaxis
fuente