生成器与yield

    函数使用yield关键字可以定义生成器对象。生成器是一个函数,它生成一个值的序列,以便在迭代中使用,例如:

def countdown(n):
    print "Counting down from %d" % n
    while n > 0:
        yield n
        n -= 1
    return

    如果调用该函数,就会发现其中的代码不会开始执行,例如:

>>> c = countdown(10)
>>> 

    相反它会返回一个生成器对象,接着,该生成器对象就会在next()(在Python3中是__next__())被调用时执行函数。例如:

>>> c.next()    #在Python3中使用c.__next__()
Counting down from 10
10
>>> c.next()
9

    调用next()时,生成器函数将不断执行语句,直至遇到yield语句为止。yield语句在函数执行停止的地方生成一个结果,直到再次调用next()。然后继续执行yield之后的语句。
    通常不会在生成器上直接调用next()方法,而是在for语句、sum()或一些使用序列的其他操作中使用它,例如

for n in countdown(10):
    statements
a = sum(countdown(10))

    生成器函数完成的标志是返回或引发StopIteration异常,这标志着迭代的结束。如果生成器在完成时返回None之外的值,都是不合法的。
    关于这个例子中的生成器还有一个有趣的问题,即生成器函数仅被部分使用。例如,考虑以下代码:

for n in countdown(10):
    if n == 2 : break
    statements

    在这个例子中,通过调用break退出for循环,而相关的生成器也没有全部完成。为了处理这种情况,生成器对象提供方法close()标识关闭。不再使用或删除生成器时,就会调用close()方法。通常不必手动调用close()方法, 但也可以怎么做,例如:

>>> c = countdown(10)
>>> c.next()
Counting down from 10
10
>>> c.next()
9
>>> c.close()
>>> c.next()
Traceback (most recent call last):
    File "<stdin>", line 1, in <module>
StopIteration
>>>

    在生成器函数内部,在yield语句上出现GeneratorExit异常时就会调用close()方法。可以选择捕捉这个异常,以便执行清理操作。

def countdown(n):
    print "Counting down from %d" % n
    try:
        while n > 0:
            yield n
            n = n - 1
    except GeneratorExit:
        print "Only made it to %d" % n

    虽然可以捕捉GeneratorExit异常,但生成器函数处理异常并使用yield语句生成另一个输出值是不合法的。另外,如果程序当前正在通过生成器进行迭代,决不能通过单独的执行线程或信号处理程序在该生成器上异步地调用close()方法。

摘自:Python参考手册第4版 6.6节 83页
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值