Comment savoir si un générateur est vide dès le départ ?

Question

Existe-t-il un moyen simple de tester si le générateur n'a pas d'éléments, comme par exemple peek , hasNext , isEmpty ou quelque chose de ce genre ?

Answer 1

Suggestion :

def peek(iterable):
    try:
        first = next(iterable)
    except StopIteration:
        return None
    return first, itertools.chain([first], iterable)

Utilisation :

res = peek(mysequence)
if res is None:
    # sequence is empty.  Do stuff.
else:
    first, mysequence = res
    # Do something with first, maybe?
    # Then iterate over the sequence:
    for element in mysequence:
        # etc.

Answer 2

La réponse simple à votre question : non, il n'y a pas de moyen simple. Il existe un grand nombre de solutions de contournement.

Il ne devrait pas y avoir de moyen simple, étant donné ce que sont les générateurs : un moyen de produire une séquence de valeurs sans garder la séquence en mémoire . Il n'y a donc pas de retour en arrière.

Vous pourriez écrire une fonction has_next ou même la glisser dans un générateur en tant que méthode avec un décorateur fantaisie si vous le souhaitiez.

Answer 3

Une façon simple est d'utiliser le paramètre optionnel de l'option suivant() qui est utilisé si le générateur est épuisé (ou vide). Par exemple :

iterable = some_generator()

_exhausted = object()

if next(iterable, _exhausted) == _exhausted:
    print('generator is empty')

Edit : Corrigé le problème signalé dans le commentaire de mehtunguh.

Answer 4

La meilleure approche, à mon avis, serait d'éviter un test spécial. La plupart du temps, l'utilisation d'un générateur est le test :

thing_generated = False

# Nothing is lost here. if nothing is generated, 
# the for block is not executed. Often, that's the only check
# you need to do. This can be done in the course of doing
# the work you wanted to do anyway on the generated output.
for thing in my_generator():
    thing_generated = True
    do_work(thing)

Si cela ne suffit pas, vous pouvez toujours effectuer un test explicite. A ce stade, thing contiendra la dernière valeur générée. Si rien n'a été généré, elle sera indéfinie - à moins que vous n'ayez déjà défini la variable. Vous pouvez vérifier la valeur de thing mais ce n'est pas très fiable. Au lieu de cela, il suffit de définir un drapeau dans le bloc et de le vérifier par la suite :

if not thing_generated:
    print "Avast, ye scurvy dog!"

Answer 5

Je déteste offrir une deuxième solution, surtout une que je n'utiliserais pas moi-même, mais, si vous devez absolument avait de faire cela et de ne pas consommer le générateur, comme dans d'autres réponses :

def do_something_with_item(item):
    print item

empty_marker = object()

try:
     first_item = my_generator.next()     
except StopIteration:
     print 'The generator was empty'
     first_item = empty_marker

if first_item is not empty_marker:
    do_something_with_item(first_item)
    for item in my_generator:
        do_something_with_item(item)

Maintenant, je n'aime vraiment pas cette solution, car je crois que ce n'est pas ainsi que les générateurs doivent être utilisés.