J'ai une fonction de générateur comme celle-ci:

def myfunct():
  ...
  yield result

La façon habituelle d'appeler cette fonction serait:

for r in myfunct():
  dostuff(r)

Ma question, existe-t-il un moyen d'obtenir un seul élément du générateur quand je le souhaite? Par exemple, j'aimerais faire quelque chose comme:

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

Créer un générateur en utilisant

g = myfunct()

Chaque fois que vous souhaitez un article, utilisez

next(g)

(ou g.next()en Python 2.5 ou inférieur).

Si le générateur sort, il augmentera StopIteration. Vous pouvez soit intercepter cette exception si nécessaire, soit utiliser l' defaultargument pour next():

next(g, default_value)

Pour choisir un seul élément d'un générateur à utiliser breakdans une forinstruction, oulist(itertools.islice(gen, 1))

Selon votre exemple (littéralement), vous pouvez faire quelque chose comme:

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

Si vous voulez " obtenir un seul élément du générateur [une fois généré] chaque fois que je le souhaite " (je suppose que 50% c'est l'intention d'origine et l'intention la plus courante) alors:

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

De cette façon, l'utilisation explicite de generator.next()peut être évitée, et la gestion de fin d'entrée ne nécessite pas de StopIterationgestion d'exceptions (cryptique) ou de comparaisons de valeurs par défaut supplémentaires.

La section d'instructions else:of forn'est nécessaire que si vous voulez faire quelque chose de spécial en cas de fin de générateur.

Remarque sur next()/ .next():

En Python3, la .next()méthode a été renommée pour .__next__()une bonne raison: elle est considérée comme de bas niveau (PEP 3114). Avant Python 2.6, la fonction intégrée next()n'existait pas. Et il a même été discuté de passer next()au operatormodule (ce qui aurait été judicieux), en raison de son besoin rare et de l'inflation douteuse des noms intégrés.

L'utilisation next()sans défaut est toujours une pratique de très bas niveau - jetant StopIterationouvertement le cryptique comme un boulon dans le code d'application normal. Et l'utilisation next()avec la sentinelle par défaut - qui devrait être la meilleure option pour une entrée next()directe builtins- est limitée et donne souvent raison à une logique / lisibilité non pythonique étrange.

Bottom line: L'utilisation de next () devrait être très rare - comme l'utilisation des fonctions du operatormodule. En utilisant for x in iterator, islice, list(iterator)et d' autres fonctions acceptant un itérateur est en toute transparence la façon naturelle d'utiliser itérateurs au niveau de l' application - et tout à fait toujours possible. next()est de bas niveau, un concept supplémentaire, non évident - comme le montre la question de ce fil. Tandis que, par exemple, l'utilisation breakdans forest conventionnelle.

Je ne pense pas qu'il existe un moyen pratique de récupérer une valeur arbitraire à partir d'un générateur. Le générateur fournira une méthode next () pour se déplacer, mais la séquence complète n'est pas produite immédiatement pour économiser de la mémoire. C'est la différence fonctionnelle entre un générateur et une liste.

Pour ceux d'entre vous qui parcourent ces réponses pour un exemple de travail complet pour Python3 ... et bien voilà:

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)

Cela produit:

1001
1002
1003

Le générateur est une fonction qui produit un itérateur. Par conséquent, une fois que vous avez une instance d'itérateur, utilisez next () pour extraire l'élément suivant de l'itérateur. À titre d'exemple, utilisez la fonction next () pour récupérer le premier élément, puis utilisez-la for inpour traiter les éléments restants:

# 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()

assurez-vous d'attraper l'exception levée après la prise du dernier élément

Je crois que la seule façon est d'obtenir une liste de l'itérateur puis d'obtenir l'élément que vous voulez de cette liste.

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