Python生成器(yield)

对于调用一个普通的Python函数,一般是从函数的第一行代码开始执行,结束于return语句、异常或者函数所有语句执行完毕。一旦函数将控制权交还给调用者,就意味着全部结束。函数中做的所有工作以及保存在局部变量中的数据都将丢失。再次调用这个函数时,一切都将从头创建。Python是通过生成器来实现类似于协同程序的概念:生成器可以暂时挂起函数,并保留函数的局部变量等数据,然后在再次调用它的时候,从上次暂停的位置继续执行下去。

提高你的 Python:解释 yield 和 Generators(生成器)

>>> a = [i for i in range(100) if not(i%2) and i%3]
>>> a
[2, 4, 8, 10, 14, 16, 20, 22, 26, 28, 32, 34, 38, 40, 44, 46, 50, 52, 56, 58, 62, 64, 68, 70, 74, 76, 80, 82, 86, 88, 92, 94, 98]
>>> b = {i:i % 2 == 0 for i in range(10)}
>>> b
{0: True, 1: False, 2: True, 3: False, 4: True, 5: False, 6: True, 7: False, 8: True, 9: False}
>>> c = {i for i in [2,1,1,1,5,6,9,0,3,4,5,5,7,6,8,3]}
>>> c
{0, 1, 2, 3, 4, 5, 6, 7, 8, 9}
>>> d = "i for i in 'I love fishc.com'"
>>> d
"i for i in 'I love fishc.com'"
>>> e = (i for i in range(10))
>>> e
<generator object <genexpr> at 0x0000000002F08AF8>
>>> next(e)
0
>>> next(e)
1
>>> next(e)
2
>>> next(e)
3
>>> for each in e:
    print(each)

    
4
5
6
7
8
9
>>> sum(i for i in range(100))
4950

以上是列表推导式,其实还有字典推导式,而没有元组推导式,元组推导式的结果其实是一个生成器,如上例的e。

def fibs():
    a = 0
    b = 1
    while True:
        a, b = b, a + b
        yield a

        
>>> for each in fibs():
    if each > 100:
        break
    print(each, end=' ')

    
1 1 2 3 5 8 13 21 34 55 89 

上面的代码就是通过生成器来实现的斐波拉契数列。生成器是遇到yield关键字时,函数就可以暂时挂起的,只有当重新访问生成器的时候,函数又会从生成器的上次的地方继续执行。

posted @ 2016-01-21 22:00  翌逍  阅读(1782)  评论(0编辑  收藏  举报