生成器与yield

函数使用yield关键字可以定义生成器对象。

生成器是一个函数,它生成一个值的序列,以便在迭代中使用,例如:

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

如果调用该函数,就会发现其中的代码不会开始执行,相反它会返回一个生成器对象。接着,该生成器对象就会在next()被调用(python3里是__next__())时执行函数。

通常不会在生成器上直接调用__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()方法,但也可以这么做

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

def countdown(n):
    print("counting down from %d" %n)
    try:
        while n > 0:
            yield n
            n -= 1
    except GeneratorExit:
        print("do some clean job")

虽然可以捕获GeneratorExit异常,但生成器函数处理异常并使用yield语句生成另一个输出值是不合法的。

另外,如果程序当前正通过生成器进行迭代,决不能通过单独的执行线程或信号处理程序在该生成器上异步地调用close()方法

更多推荐

Python基础-生成器与yield