Python generator和yield的一点理解

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.下面是运行结果:


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值