comment n'en choisir qu'un seul élément à partir d'un générateur (en python)?

J'ai un générateur de fonction comme suit:

def myfunct():
  ...
  yield result

L'habitude d'appeler cette fonction serait:

for r in myfunct():
  dostuff(r)

Ma question, est-il un moyen d'obtenir qu'un élément de la génératrice à chaque fois que j'aime?
Par exemple, j'aimerais faire quelque chose comme:

while True:
  ...
  if something:
      my_element = pick_just_one_element(myfunct())
      dostuff(my_element)
  ...
InformationsquelleAutor Alexandros | 2011-01-19
  1. 272

    Créer un générateur à l'aide de

    g = myfunct()

    Chaque fois que vous voulez un élément, utilisez

    next(g)

    (ou g.next() dans Python 2.5 ou ci-dessous).

    Si le générateur de sort, il va augmenter StopIteration. Vous pouvez prendre cette exception, si nécessaire, ou de l'utilisation de la default argument next():

    next(g, default_value)
    • Remarque, il ne fera que soulever StopIteration lorsque vous essayez d'utiliser.g.next() après le dernier élément de g a été fourni.
    • next(gen, default) peut également être utilisé pour éviter les StopIteration exception. Par exemple next(g, None) pour un générateur de cordes, soit le rendement d'une chaîne ou d'Aucun après l'itération était fini.
    • en Python 3000, next () __next__()
    • Votre commentaire est quelque peu trompeur. En Python 3, vous pouvez utiliser la deuxième syntaxe donnée dans ma réponse, next(g). Cela va à l'interne appel g.__next__(), mais vous n'avez pas vraiment à vous soucier de qui, tout comme vous n'avez généralement pas de soins que len(a) appelle en interne a.__len__().
    • Je devrais avoir été plus clair. g.next() est g.__next__() dans py3k. Le groupe builtin next(iterator) a été autour depuis la version 2.6 de Python, et c'est ce que devrait être utilisé dans tous les nouveaux code Python, et il est trivial de backimplement si vous avez besoin de soutien py <= 2.5.
    • Je pense que @kxr fournit de bonnes raisons (cinq ans plus tard, d'où le manque de upvotes) pour pas explicitement la peine avec next() et la StopIteration exception dans leur excellente réponse. itertools.islice() est un vrai bijou, joli et Pythonic, et il ne faut pas avoir peur de la boucle for en faveur des one-liners.
    • Ces "bonnes raisons" qu'il est "bizarre", "rarement" et "très faible niveau de pratique" et "non-pythonic". Il est facile de prouver que "rarement utilisé" est erroné. Les autres "raisons" sont simplement énoncés, sans aucune référence ou de la justification, et je ne peux pas dire comment kxr sont venus à ces conclusions. À mon avis, le haut- next() est fourni pour exactement cet effet, donc c'est la fonction naturelle à utiliser, et c'est ce que la plupart des gens utilisent à cette fin.
    • La déclaration est que "l'Utilisation de la fonction next() doit très rare" (mon emphase). C'est une question d'interprétation, mais je pense que ce n'est pas une observation de l'utilisation effective, mais sur les meilleures pratiques. Quoi de plus pythonic est une question de goût, bien sûr. Pour ma part, je suis d'accord avec @kxr qu'en utilisant le protocole de niveau supérieur qui fait pour le moins obscur code, même si next() n'est pas que obscures. Je suis reconnaissant pour le conseil à propos de l'existence de islice(), ce qui était exactement ce que je cherchais quand j'ai trouvé cette question.

    InformationsquelleAutor Sven Marnach
  2. 26

    Pour la cueillette qu'un élément d'un générateur d'utilisation break dans un for déclaration, ou list(itertools.islice(gen, 1))

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

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

    Si vous voulez "obtenir juste un élément de la [une fois généré] générateur à chaque fois que j'aime" (je suppose que 50% c'est l'intention d'origine, et la plus commune intention) puis:

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

    De cette façon explicite l'utilisation de generator.next() peuvent être évités, et de fin de gestion des données ne nécessite pas (cryptique) StopIteration la gestion des exceptions ou extra-valeur par défaut des comparaisons.

    La else: de for section sur la déclaration n'est nécessaire que si vous voulez faire quelque chose de spécial en cas de fin de générateur.

    Note sur next() /.next():

    En Python3 le .next() méthode a été renommé .__next__() pour une bonne raison: son faible niveau de PEP (3114). Avant de Python 2.6 la fonction builtin next() n'existe pas. Et il a même été discutée à déplacer next() à la operator module (ce qui aurait été sage), en raison de sa rareté du besoin et discutable de l'inflation de builtin noms.

    À l'aide de next() sans défaut est encore très faible au niveau de la pratique de jeter le cryptique StopIteration comme un boulon dans le bleu normal le code de l'application ouvertement. Et à l'aide de next() par défaut sentinel - la meilleure qui doit être la seule option pour un next() directement dans builtins - est limitée et donne souvent raison à bizarre non-pythonic logique/readablity.

    Ligne du bas: Utilisation de la fonction next() doit être très rare - comme l'utilisation de fonctions de operator module. À l'aide de for x in iterator , islice, list(iterator) et d'autres fonctions de l'acceptation d'un itérateur de façon transparente est la façon naturelle de l'utilisation des itérateurs sur le niveau d'application - et pas toujours possible. next() est de bas niveau, un concept, non évidente - comme la question de ce fil de discussion montre. Alors que par exemple à l'aide de break dans for est classique.

    • Ce travail est beaucoup trop juste pour obtenir le premier élément d'une liste de résultat. Assez souvent, je n'ai pas besoin d'être paresseux, mais n'ont pas de choix dans py3. N'est-il pas quelque chose qui s'apparente à mySeq.head ?
    InformationsquelleAutor kxr
  3. 2

    Je ne crois pas qu'il y est 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 parcourir lui-même, mais l'ensemble de la séquence n'est pas produit immédiatement pour économiser de la mémoire. C'est la différence de fonctionnement entre un générateur et une liste.

    InformationsquelleAutor g.d.d.c
  4. 0
    generator = myfunct()
while True:
   my_element = generator.next()

    assurez-vous d'attraper l'exception levée après le dernier élément est pris

    • Non valable pour Python 3, voir l'excellent réponse par kxr.
    • Il suffit de remplacer "générateur.next()" avec "suivant(générateur)" pour Python 3.
    InformationsquelleAutor John
  5. 0

    Générateur est une fonction qui produit un itérateur. Par conséquent, une fois que vous avez itérateur exemple, l'utilisation next() pour récupérer l'élément suivant de l'itérateur.
    Par exemple, utiliser la fonction next() pour récupérer le premier élément, et de les utiliser plus tard for in pour 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(item)
print('first item:', first)

# process remaining items:
for item in items:
    print('next item:', item)
    InformationsquelleAutor andrii
  6. -3

    Je crois que c'est le seul moyen pour obtenir une liste à partir de l'itérateur, puis obtenir l'élément que vous voulez à partir de cette liste.

    l = list(myfunct())
l[4]
    • Sven la réponse est sans doute mieux, mais je vais juste laisser ça ici au cas où c'est plus en ligne avec vos besoins.
    • Assurez-vous que vous avez un fini générateur avant de le faire.
    • Désolé, c'est la complexité de la longueur de l'itérateur, alors que le problème est évidemment O(1).
    • Perdre trop de mémoire et des processus à sucer à partir du générateur! Aussi, comme @Seth mentionné avant, les générateurs ne sont pas garantis, quand arrêter de générer.
    • C'est évidemment pas, le seul moyen (ou la meilleure façon si myfunct() génère un grand nombre de valeurs), puisque vous pouvez utiliser la fonction intégrée next pour obtenir la prochaine valeur générée.
    InformationsquelleAutor keegan3d