GeeksforGeeks App
Open App
Browser
Continue

When to use yield instead of return in Python?

The yield statement suspends a function’s execution and sends a value back to the caller, but retains enough state to enable the function to resume where it left off. When the function resumes, it continues execution immediately after the last yield run. This allows its code to produce a series of values over time, rather than computing them at once and sending them back like a list.

Let’s see with an example:

Python

 `# A Simple Python program to demonstrate working``# of yield` `# A generator function that yields 1 for the first time,``# 2 second time and 3 third time`  `def` `simpleGeneratorFun():``    ``yield` `1``    ``yield` `2``    ``yield` `3`  `# Driver code to check above generator function``for` `value ``in` `simpleGeneratorFun():``    ``print``(value)`

Output:

```1
2
3```

Return sends a specified value back to its caller whereas Yield can produce a sequence of values. We should use yield when we want to iterate over a sequence, but don’t want to store the entire sequence in memory. Yield is used in Python generators. A generator function is defined just like a normal function, but whenever it needs to generate a value, it does so with the yield keyword rather than return. If the body of a def contains yield, the function automatically becomes a generator function.

Python

 `# A Python program to generate squares from 1``# to 100 using yield and therefore generator` `# An infinite generator function that prints``# next square number. It starts with 1`  `def` `nextSquare():``    ``i ``=` `1` `    ``# An Infinite loop to generate squares``    ``while` `True``:``        ``yield` `i``*``i``        ``i ``+``=` `1`  `# Next execution resumes``        ``# from this point`  `# Driver code to test above generator``# function``for` `num ``in` `nextSquare():``    ``if` `num > ``100``:``        ``break``    ``print``(num)`

Output:

```1
4
9
16
25
36
49
64
81
100```

My Personal Notes arrow_drop_up