python生成器

生成器

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

生成器与return的区别:

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

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

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

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

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

1
2
3
4
5
6
7
8
9
10
11
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)

输出结果为:

1
2
3
4
5
6
7
start games
5
4
3
2
1
game over