生成器中yield 与 return

这里用的是python 3.6

可以用yield定义生成器,生成器保存的是算法,每迭代一次,返回一个yield后的值,直到遇到StopIteration,迭代完成,即next指向StopIteration,这个生成器不能再次被迭代

初学不能理解yield的时候,就把yield看成print,只不过print是返回给人看的

,yield是返回给机器的

但我们在用yield定义生成器的时候,又写了return会怎么样

# 遍历这个生成器,遇到return,停止遍历,
# 在这里return就是StopIteration
def g2():
    yield 'a'
    yield 'b'
    yield 'c'
    return
    yield 'd'

for n in g2():
    print(n)
# 这个return在我完全迭代完之后,
# 所以看上去这个return没影响
def fib(n):
    a, b = 0, 1
    while(n>0):
        yield a
        a, b = b, a+b
        n-=1
    return a

猜你喜欢

转载自wzgdavid.iteye.com/blog/2388632