Python – How to know if a generator is empty from the start

generatorpython

Is there a simple way of testing if the generator has no items, like peek, hasNext, isEmpty, something along those lines?

Best Solution

Suggestion:

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

Usage:

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.