生成器
一边循环一边计算出后续元素的机制,称为生成器:generator
创建generator的第一种方法:
把列表生成式的[ ]改为( ):
>>>L=[x*x for x in range(10)]
>>>L
[0,1,4,9,16,25,36,49,64,81]
>>>g=(x*x for x in range(10))
>>>g
<generator object<genexpr> at ox1022ef63>
可以通过next( )函数获得generator的下一个返回值。
>>>next(g)
0
>>>next(g)
1
...
超出索引范围后会Stop Iteration
更高效的获取generator的方法是使用for
>>>for n in g:
print(n)
创建generator的第二种方法是yield语句
调用next()的时候执行,遇到yield语句返回,再次执行时从上一次的返回的yield语句处继续执行。
例:定义一个generator,依次返回数字1,3,5
>>>def odd():
print('step 1')
yield 1
print('step 2')
yield (3)
print('step 3')
yield(5)
>>>o=odd()
>>>next(o)
step 1
1
>>>next(o)
step 2
3
>>>next(o)
step 3
5