1.定义:在Python的函数定义中,如果出现了yield表达式,那么定义的函数就不再是一个普通的函数,而是一个generator函数,调用这个函数的返回值就叫generator。
例如:
def produce(l):
i=0
while 1:
if i < 10:
l.append(i)
yield
i=i+1
else:
return
上面定义的函数因为带有yield表达式,所以是一个generator函数,该函数可以有return,但是不能带有任何值。
2.具体使用过程分析:请看下面的代码
import time
import sys
import datetime
# 生产者
def produce(l):
print("procuce begin")
i=0
while 1:
if i < 3:
l.append(i)
print("produce stop")
yield
print("produce continue")
i=i+1
time.sleep(5)
else:
return
# 消费者
def consume(l):
p = produce(l)
while 1:
try:
print("before next")
next(p)
print("after next")
while len(l) > 0:
print(l.pop(),'\n')
except StopIteration:
sys.exit(0)
if __name__ == "__main__":
l = []
consume(l)
上面的代码首先定义了一个generator函数produce,然后在consume函数中调用了改方法,来实现每隔5秒依次打印数字0、1、2。
代码运行过程是怎样的呢?首先是p=produce(l)这里调用函数的时候,由于produce是一个generator函数,所以函数并不会立即执行,而是暂时挂起,并且返回一个generator p,通过查看p的类型,显示为<class 'generator'>,然后consume函数进入while循环,首先打印“before next”,接着调用next(p),调用之后就会produce函数就会从挂起状态开始执行,依次打印“produce begin”、“produce stop”,当遇到yield的时候,generator函数又会挂起,所以就接着返回consume函数继续执行,打印“after next”,最后打印列表l中的元素“0”,执行完成之后继续while循环,重复刚刚的运行顺序,直到next(p)抛出StopIteration异常,程序运行结束。
3.下面是运行结果: