python生成器

生成器

生成器就是一个函数，这个函数内包含yield这个关键字

生成器与return的区别：

return只能返回一次函数就结束，而生成器可以返回多次值

生成器函数包含一个或者多个yield

当调用生成器函数时，函数将返回一个对象，但是不会立刻向下执行，yield会保存一个状态，下一次运行会在这个位置接着往下运行，直到碰到下一个yield

像__iter__()和__next__()方法等是自动实现的，所以我们可以通过next()方法对对象进行迭代

一旦函数被yield，函数会暂停，控制权返回调用者

 

def pr(x):
    print("start games")
    while x > 0:
        yield x
        x -= 1
    print("game over")


game = pr(5)    #此时game就是一个迭代器
for i in game:
    print(i)

 

输出结果为：

start games

5

4

3

2

1

game over