generator.py

def square_gen(num):
    for i in range(num):
        yield i**2

gen = square_gen(5)

for i in gen:
    print(i)
    
for i in map(lambda x: x**2, range(5)):
    print(i)

squares = (x ** 2 for x in range(5))