Python: Insert element(s) at beginning of generator

user_5897987 0
Viewed 248

If I have a generator, e.g.:

g = (i for i in range(10))

and later I want to update the generator so that the first element(s) yielded is now some other element(s), what is the most "pythonic" way to do this?

Clearly, I could define a basic generator extender function such as:

def extend(generator, new_elements):
    for element in new_elements:
        yield element
    for element in generator:
        yield element

but I am wondering if this is necessary or if there is some better way to do this?

generator python
2 Answers

You could use itertools.chain for this:

>>> from itertools import chain
>>> g = (i for i in range(10))
>>> list(chain(['a','b','c'], g))
['a', 'b', 'c', 0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
user_50144550

You can use the following function:

def chain(*iterables):
    for it in iterables:
        yield from it
user_89736200
Related