初识yield表达式是在生成器学习中,只要函数中出现yield表达式,那么这个函数就是生成器函数。当函数执行到yield那一行时,yield就会返回后面的值,然后冻结当前函数状态,下行代码不执行,等下一次调用再唤醒接着往下执行。
先来个简单生成器函数:
def my_gen():
yield 1
yield 2
yield 3
用next方法获取值:
ret = my_gen()
print(next(ret))
print(next(ret))
print(next(ret))
结果是1,2,3
如果把yield表达式赋值给一个变量,打印那个变量,结果会是怎样呢?再来一个简单生成器 试试看
def my_gen(start):
while start < 10:
temp = yield start
print(temp)
start += 1
先来调用下函数,把1传给my_gen()
ret = my_gen(1) print(ret.__next__())
函数自上往下运行,进行while条件判断,1 < 10,下一条yield表达式,遇yield 返回后面的值,就是1。再把yield start的结果赋给变量temp,遇yield还会冻结函数状态,下行代码print语句不执行。
我们在调用一次next方法
print(ret.__next__())
此时函数苏醒,从print语句开始执行,先打印temp值,然后start进行+1操作,再进行条件判断,遇yield又抛出start值,现在start = 2,所以打印2。接下来又又冻结函数状态,停了下来。
本来我以为temp = 2,结果打印结果是None。顿时摸不着头脑yield明明返回的值是后面的start,怎么赋值给一个变量就变成None呢? 后来查资料,解释是第一次调用next方法后,yield先返回后面start的值出去,接下来已经没有其他值,只能赋None值给temp。