Tengo una función de generador como la siguiente:

def myfunct():
  ...
  yield result

La forma habitual de llamar a esta función sería:

for r in myfunct():
  dostuff(r)

Mi pregunta, ¿hay alguna manera de obtener solo un elemento del generador cuando lo desee? Por ejemplo, me gustaría hacer algo como:

while True:
  ...
  if something:
      my_element = pick_just_one_element(myfunct())
      dostuff(my_element)
  ...

Crea un generador usando

g = myfunct()

Cada vez que desee un artículo, use

next(g)

(o g.next()en Python 2.5 o inferior).

Si el generador sale, subirá StopIteration. Puede capturar esta excepción si es necesario, o usar el defaultargumento para next():

next(g, default_value)

Para elegir solo un elemento de un generador, use breaken una fordeclaración, olist(itertools.islice(gen, 1))

Según su ejemplo (literalmente) puede hacer algo como:

while True:
  ...
  if something:
      for my_element in myfunct():
          dostuff(my_element)
          break
      else:
          do_generator_empty()

Si desea " obtener solo un elemento del generador [una vez generado] cuando quiera " (supongo que el 50% es la intención original y la intención más común) entonces:

gen = myfunct()
while True:
  ...
  if something:
      for my_element in gen:
          dostuff(my_element)
          break
      else:
          do_generator_empty()

De esta manera, generator.next()se puede evitar el uso explícito de , y el manejo de fin de entrada no requiere (críptico)StopIteration manejo de excepciones o comparaciones de valores predeterminados adicionales.

La sección else:de fordeclaración solo es necesaria si desea hacer algo especial en caso de fin de generador.

Nota sobre next()/ .next():

En Python3 .next()se cambió el nombre del método .__next__()por una buena razón: se considera de bajo nivel (PEP 3114). Antes de Python 2.6, la función integrada next()no existía. E incluso se discutió la posibilidad de pasar next()al operatormódulo (lo que habría sido prudente), debido a su rara necesidad y la cuestionable inflación de los nombres incorporados.

Usar next()sin valor predeterminado sigue siendo una práctica de muy bajo nivel: lanzar el críptico StopIterationcomo un rayo de la nada en el código de aplicación normal abiertamente. Y usar next()con centinela predeterminado, que mejor debería ser la única opción para un next()directamente enbuiltins , es limitado y a menudo da razón a una lógica / legibilidad no pitónica extraña.

En pocas palabras: Usar next () debería ser muy raro, como usar funciones del operatormódulo. El uso de for x in iterator, islice, list(iterator)y otras funciones de aceptar un iterador sin problemas es la forma natural de usar iteradores de nivel de aplicación - y bastante siempre es posible. next()es de bajo nivel, un concepto adicional, no obvio, como lo muestra la pregunta de este hilo. Mientras que, por ejemplo, el uso breaken fores convencional.

No creo que haya una manera conveniente de recuperar un valor arbitrario de un generador. El generador proporcionará un método next () para atravesarse, pero la secuencia completa no se produce inmediatamente para ahorrar memoria. Esa es la diferencia funcional entre un generador y una lista.

Para aquellos de ustedes que escanean estas respuestas para obtener un ejemplo de trabajo completo para Python3 ... bueno, aquí van:

def numgen():
    x = 1000
    while True:
        x += 1
        yield x

nums = numgen() # because it must be the _same_ generator

for n in range(3):
    numnext = next(nums)
    print(numnext)

Esto produce:

1001
1002
1003

El generador es una función que produce un iterador. Por lo tanto, una vez que tenga una instancia de iterador, use next () para obtener el siguiente elemento del iterador. Como ejemplo, use la función next () para obtener el primer elemento, y luego use for inpara procesar los elementos restantes:

# create new instance of iterator by calling a generator function
items = generator_function()

# fetch and print first item
first = next(items)
print('first item:', first)

# process remaining items:
for item in items:
    print('next item:', item)

generator = myfunct()
while True:
   my_element = generator.next()

asegúrese de atrapar la excepción lanzada después de tomar el último elemento

Creo que la única forma es obtener una lista del iterador y luego obtener el elemento que desea de esa lista.

l = list(myfunct())
l[4]