什么事生成器
本质上也是迭代器,不过它比较特殊。
生成器创建
需要使用yield关键字
下面是个简单的生成器
例如:
def test:
yield 1
yield这个关键字其实类似于return关键字,return关键字的作用是在函数中使用,用来返回数据,yield关键字的作用也是一样的,就是用来返回数据,但是和return还有其它的不同之处。
return执行的时候,函数就跳出
yield执行的时候,返回数据,但是函数就会记住跳出的位置,当你再次调用函数(生成器)的时候,就从上一次跳出的地方继续执行
改变一下迭代器的代码,代码简洁多了
用next()
方法取值
def test():
a = 0
while 1:
a = a+1
yield a
t = test()
print(next(t))
print(next(t))
print(next(t))
结果:
1
2
3
for循环取值
for循环会自动调用next()方法
def test():
a = 0
while 1:
a = a+1
yield a
t = test()
for i in t:
print(i)
结果是a的值一直在增加