近期在使用python中遇到了yield表达式,不是很明白其含义。这里写下这篇文章谈谈自己对yield的理解。
当函数中具有yield表达式时,这个函数就是一个可迭代生成器。
# 演示创建一个可迭代生成器函数
def func(n:int):
while n > 0:
yield n
print(n)
n = n-1
# 首次调用该函数时返回的是一个生成器
print(func(3))

这点是和普通函数区别较大的地方!
首次使用返回的是生成器对象,而不是3!
yield表达式所在的地方,具有记忆作用。每次运行时都只执行到yield后的部分,并记录此时的变量值。当下一次调用时,在从上一次运行结束的地方继续运行,直到遇到yield结束。
gen = func(3)
first = next(gen)
print("*" * 20)
second = next(gen)
print("*" * 20)
third = next(gen)
print("*" * 20)
forth = next(gen)

结果分析:
gen = func(3) 取得可迭代生成器的对象。
first = next(gen) 第一次执行生成器对象内部逻辑。此时函数内部逻辑只运行到yield n 部分,
所以这一步时没有打印3。此时n为3
print(“*” * 20) 打印用来分割第一次和第二次执行生成器对象内部逻辑。
second = next(gen) 第二次执行生成器对象内部逻辑。此时函数内部逻辑从print(n) 部分开始执行,直到yield n 部分结束。此时n为2
==print("" * 20)== 打印用来分割第二次和第三次执行生成器对象内部逻辑。
third = next(gen) 第三次执行生成器对象内部逻辑。此时函数内部逻辑从print(n) 部分开始执行,直到yield n 部分结束。此时n为1
==print("" * 20)== 打印*用来分割第三次和第四次执行生成器对象内部逻辑。
forth = next(gen) 第四次执行生成器对象内部逻辑。此时函数内部逻辑从print(n) 部分开始执行,经过n = n - 1 指令后,n为0。此时n不满足while条件无法执行到yield n 部分,故产生StopIteration 异常。
print("-"*20)
gen1 = func(3)
for i in gen1:
print(i)
当用在for循环中时i是每次yield n之后的n。