How to perform lazy evaluation in Python?

In Python, lazy evaluation refers to the practice of deferring the evaluation of an expression until its result is actually needed. This can help optimize performance and conserve memory by avoiding unnecessary computations.

There are several ways to achieve lazy evaluation in Python:

  1. Generator Functions: Generator functions are defined using the yield keyword instead of return. They generate values on-the-fly as they are requested, allowing for lazy evaluation. When a generator function is called, it returns a generator object, which can be iterated over. The generator function's body is only executed when the values are requested, and it pauses and resumes its execution state using the yield keyword.


def lazy_evaluation():
    for i in range(5):
        yield i ** 2

generator = lazy_evaluation()

for value in generator:

In this example, the lazy_evaluation() generator function generates squared values on-the-fly. The values are computed and yielded one at a time, only when requested during iteration.

  1. Generator Expressions: Generator expressions are similar to list comprehensions but use parentheses instead of square brackets. They create generator objects that produce values lazily, on-demand, rather than storing all the values in memory at once. Generator expressions are particularly useful when you need to generate large or infinite sequences.


generator = (i ** 2 for i in range(5))

for value in generator:

In this example, the generator expression (i ** 2 for i in range(5)) generates squared values on-the-fly, similar to the previous example with the generator function.

  1. Lazy Evaluation Libraries: Python also has several libraries that provide lazy evaluation capabilities, allowing you to define and compose lazy computations. Examples include itertools and toolz, which provide functions like map(), filter(), and accumulate() that operate lazily on iterables.


from itertools import count, islice

generator = map(lambda x: x ** 2, count())

for value in islice(generator, 5):

In this example, map() and count() from the itertools module are used to generate an infinite sequence of squared values. The islice() function is used to limit the output to the first 5 values.

By using generator functions, generator expressions, or lazy evaluation libraries, you can implement lazy evaluation in Python and defer the computation of values until they are actually needed. This approach can help optimize performance and memory usage, especially when working with large or infinite sequences.

Related Questions You Might Be Interested