yield的直观理解是把它看作return,即可从函数中返回值,但其与return的区别是:return返回后语句结束,而yield可执行多次返回操作,直到函数所在语句结束。含有yield关键字的函数为生成器。
问题:假设我们要从0至100的正整数中返回5的倍数。
方法1:
def my_function():
for i in range(100):
if i%5 == 0:
print(i)
my_function()
利用print函数打印结果,缺点是可复用性差,其他函数无法获得该函数的结果。
方法2:
def my_function():
L = []
for i in range(100):
if i%5 == 0:
L.append(i)
return L
L = my_function()
利用列表存储数据,解决了print函数可复用性差的问题,然而列表占用内存随着个数的增大而增大。
方法3:
def my_function():
for i in range(100):
if i%5 == 0:
yield i
for i in my_function():
print(i)
利用yield将函数变成生成器函数,调用该函数会返回一个可迭代对象,在for循环执行时,每次循环都会执行函数内部代码,执行到yield i 时就返回一个迭代值并中断执行。下次迭代时,代码就从函数内部的yield i 的下一条语句继续执行,而函数的本地变量看起来和上次中断执行前是完全一样的,于是函数继续执行,直到再次遇到 yield i,如此反复执行 。与用列表存储相比,占用内存较少。
含有yield的函数的生成器对象,也含有next方法,使用方式如下:
def my_function():
for i in range(100):
if i%5 == 0:
yield i
f = my_function()
print(f.__next__()) # 返回 0
print(f.__next__()) # 返回 5
print(f.__next__()) # 返回 10