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
Author by
Walrus the Cat
Updated on June 24, 2022Comments
-
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.