Question

Write a function to generate cube's of numbers over time.

Answer 1

A generator can produce cubes one at a time instead of storing the entire sequence in memory.

def cubes(start, stop, step=1):
    if step == 0:
        raise ValueError("step cannot be zero")

    number = start
    if step > 0:
        while number <= stop:
            yield number ** 3
            number += step
    else:
        while number >= stop:
            yield number ** 3
            number += step


for cube in cubes(1, 5):
    print(cube)

Output:

1
8
27
64
125