First page Back Continue Last page Overview Image

Splitting up a Generator

True

def cities():

for city in ["Berlin", "Hamburg", "Munich", "Freiburg"]:

yield city

def squares():

for number in range(10):

yield number ** 2

def generator_all_in_one():

for city in cities():

yield city

for number in squares():

yield number

def generator_splitted():

yield from cities()

yield from squares()

lst1 = [el for el in generator_all_in_one()]

lst2 = [el for el in cities()] + [el for el in squares()]

print(lst1 == lst2)