Is there any shorthand for 'yield all the output from a generator'?

11,251

Solution 1

In Python 3.3+, you can use yield from. For example,

>>> def get_squares():
...     yield from (num ** 2 for num in range(10))
... 
>>> list(get_squares())
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

It can actually be used with any iterable. For example,

>>> def get_numbers():
...     yield from range(10)
... 
>>> list(get_numbers())
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
>>> def get_squares():
...     yield from [num ** 2 for num in range(10)]
... 
>>> list(get_squares())
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

Unfortunately, Python 2.7 has no equivalent construct :'(

Solution 2

You can use a list comprehension to get all of the elements out of a generator (assuming the generator ends):

[x for x in generator]

Solution 3

Here is a simple one-liner valid in Python 2.5+ as requested ;-):

for thing in generator: yield thing
Share:
11,251
Walrus the Cat
Author by

Walrus the Cat

Updated on June 24, 2022

Comments

  • Walrus the Cat
    Walrus the Cat almost 2 years

    Is there a one-line expression for:

    for thing in generator:
        yield thing
    

    I tried yield generator to no avail.