python生成器

#(如果一个函数定义中包含yield关键字，那么这个函数就不再是一个普通函数，而是一个generator)
g = (x * x for x in range(2,10))#创建生成器（()表示生成器、[]列表解析）
print(next(g))#只打印第一个元素
for i in g:#通过循环遍历元素
    print(i)

def fib(max):
    n, a, b = 0, 0, 1
    while n < max:
        yield b
        a, b = b, a + b
        n = n + 1
    return 'done'
 
f = fib(10)
print('fib(10):', f)
for x in f:
    print(x)
 
# call generator manually:
g = fib(5)
while 1:
    try:
        x = next(g)
        print('g:', x)
    except StopIteration as e:
        print('Generator return value:', e.value)
        break

def odd():
#如果函数中有yield，那么该函数不在是普通函数，而是生成器：generator。
    print('step 1')
    yield 1
    print('step 2')
    yield(3)
    print('step 3')
    yield(5)
o = odd()
#print(o)print(next(o))
for i in o:#遍历结果，不然只显示第一条
    print(i)

 

 

转载于:https://www.cnblogs.com/chen-xia/articles/8126005.html