generators.py

import sys

# List comprehension
list_comprehension = [i for i in range(10_000_000)]
print(f"List comprehension memory: {sys.getsizeof(list_comprehension) / (1024 * 1024)} MB")

# Yield generator
def generator():
    for i in range(10_000_000):
        yield i

gen = generator()
print(f"Generator memory: {sys.getsizeof(gen) / (1024 * 1024)} MB")