How to Use Yield Keyword for Memory Efficient Python Code
Last Updated :
18 Mar, 2024
We are given a task to understand how to use the yield keyword for memory-efficient Python code with the help of different examples with their code. In this article, we will see how to use the yield keyword for memory-efficient Python code.
What is the Yield Keyword?
The yield
keyword in Python is used in the context of generator functions and is fundamental to creating iterators efficiently. It allows a function to pause its execution and yield a value to the caller without losing its state. When the function is called again, it resumes execution from where it left off.
Use Yield Keyword for Memory-Efficient Python Code
Below are some of the examples by which we can understand how to use the yield keyword for memory-efficient Python code:
Memory-Efficient Even Number Filtering
In this example, filter_even_numbers
function utilizes the yield
keyword to lazily generate even numbers from the input data, resulting in memory-efficient iteration and printing of the filtered values.
Python3
def filter_even_numbers(data):
for value in data:
if value % 2 == 0:
yield value
my_data = [1, 2, 3, 4, 5]
result_generator = filter_even_numbers(my_data)
for value in result_generator:
print(value)
Generating Lazy Sequences Efficiently
In this example, in below code The lazy_sequence
generator function produces a sequence of doubled values up to the specified limit, demonstrating memory-efficient iteration when consumed in a for loop.
Python3
def lazy_sequence(limit):
for i in range(limit):
yield i * 2
# Using the generator
gen = lazy_sequence(5)
for value in gen:
print(value)
Memory-Efficient Fibonacci Sequence Generation
In this example, the fibonacci_sequence
generator function yields Fibonacci sequence values up to the specified limit, showcasing memory-efficient iteration for generating and printing the sequence.
Python3
def fibonacci_sequence(limit):
a, b = 0, 1
while a < limit:
yield a
a, b = b, a + b
fibonacci_gen = fibonacci_sequence(50)
for value in fibonacci_gen:
print(value)
Output0
1
1
2
3
5
8
13
21
34
Memory-Efficient Task Parallelism
In this example, the parallel_tasks
generator yields lambda functions representing parallel tasks, enabling memory-efficient execution of each task on a given input value (data
in this case).
Python3
def parallel_tasks():
yield lambda x: x * 2
yield lambda x: x ** 2
# Add more tasks as needed
# Using the generator for parallel tasks
tasks_gen = parallel_tasks()
data = 5
for task in tasks_gen:
result = task(data)
print(result)
Memory Efficiency Comparison: Yield vs. Generator
In this example, below code compares memory usage between a list-based approach (without yield
) and a generator-based approach (with yield
) for generating a sequence of squares up to a specified limit. The generator-based approach demonstrates superior memory efficiency, especially for large limits, as it generates values on-the-fly without storing the entire sequence in memory.
Python3
# Without Yield (List-Based Approach)
import sys
def generate_squares_list(limit):
squares = [num ** 2 for num in range(limit)]
return squares
# With Yield (Generator-Based Approach)
def generate_squares_yield(limit):
for num in range(limit):
yield num ** 2
# Memory Comparison
limit = 10**6 # Set a large limit for demonstration purposes
# Without Yield (List-Based Approach)
squares_list = generate_squares_list(limit)
memory_usage_list = sys.getsizeof(squares_list)
# With Yield (Generator-Based Approach)
squares_yield = generate_squares_yield(limit)
memory_usage_yield = sys.getsizeof(squares_yield)
print(f"Memory usage without yield: {memory_usage_list} bytes")
print(f"Memory usage with yield: {memory_usage_yield} bytes")
OutputMemory usage without yield: 8697472 bytes
Memory usage with yield: 128 bytes
Conclusion
In conclusion, leveraging generators in Python is a powerful strategy for achieving memory-efficient code, particularly in situations where handling large datasets or dynamic data streams is essential. The lazy evaluation approach of generators allows for on-the-fly generation of values, reducing the overall memory footprint and enabling the processing of data in a scalable and efficient manner.
Similar Reads
Python Tutorial - Learn Python Programming Language Python is one of the most popular programming languages. Itâs simple to use, packed with features and supported by a wide range of libraries and frameworks. Its clean syntax makes it beginner-friendly. It'sA high-level language, used in web development, data science, automation, AI and more.Known fo
10 min read
Python Interview Questions and Answers Python is the most used language in top companies such as Intel, IBM, NASA, Pixar, Netflix, Facebook, JP Morgan Chase, Spotify and many more because of its simplicity and powerful libraries. To crack their Online Assessment and Interview Rounds as a Python developer, we need to master important Pyth
15+ min read
Non-linear Components In electrical circuits, Non-linear Components are electronic devices that need an external power source to operate actively. Non-Linear Components are those that are changed with respect to the voltage and current. Elements that do not follow ohm's law are called Non-linear Components. Non-linear Co
11 min read
Python OOPs Concepts Object Oriented Programming is a fundamental concept in Python, empowering developers to build modular, maintainable, and scalable applications. By understanding the core OOP principles (classes, objects, inheritance, encapsulation, polymorphism, and abstraction), programmers can leverage the full p
11 min read
Python Projects - Beginner to Advanced Python is one of the most popular programming languages due to its simplicity, versatility, and supportive community. Whether youâre a beginner eager to learn the basics or an experienced programmer looking to challenge your skills, there are countless Python projects to help you grow.Hereâs a list
10 min read
Python Exercise with Practice Questions and Solutions Python Exercise for Beginner: Practice makes perfect in everything, and this is especially true when learning Python. If you're a beginner, regularly practicing Python exercises will build your confidence and sharpen your skills. To help you improve, try these Python exercises with solutions to test
9 min read
Python Programs Practice with Python program examples is always a good choice to scale up your logical understanding and programming skills and this article will provide you with the best sets of Python code examples.The below Python section contains a wide collection of Python programming examples. These Python co
11 min read
Spring Boot Tutorial Spring Boot is a Java framework that makes it easier to create and run Java applications. It simplifies the configuration and setup process, allowing developers to focus more on writing code for their applications. This Spring Boot Tutorial is a comprehensive guide that covers both basic and advance
10 min read
Python Introduction Python was created by Guido van Rossum in 1991 and further developed by the Python Software Foundation. It was designed with focus on code readability and its syntax allows us to express concepts in fewer lines of code.Key Features of PythonPythonâs simple and readable syntax makes it beginner-frien
3 min read
Python Data Types Python Data types are the classification or categorization of data items. It represents the kind of value that tells what operations can be performed on a particular data. Since everything is an object in Python programming, Python data types are classes and variables are instances (objects) of thes
9 min read