简单地讲,yield 的作用就是把一个函数变成一个 generator,带有 yield 的函数不再是一个普通函数,Python 解释器会将其视为一个 generator,调用 fab(5) 不会执行 fab 函数,而是返回一个 iterable 对象!在 for 循环执行时,每次循环都会执行 fab 函数内部的代码,执行到 yield b 时,fab 函数就返回一个迭代值,下次迭代时,代码从 yield b 的下一条语句继续执行,而函数的本地变量看起来和上次中断执行前是完全一样的,于是函数继续执行,直到再次遇到 yield。
实例看下面的斐波那契(伪)数列:
它是第四个数等于前三个数之和。我叫它——斐波那契他哥数列。
比如,0, 1, 1, 2, 4, 7…
现在,希望可以写一个函数,可以得到长度为n的斐波那契他哥数列。要求,必须用生成器搞定。
def fib_bro(n):
a, b, c, count = 0, 1, 1, 0
while count < n:
yield a
# 每次算一个,返回一个值在一个类似于列表的东西里面(迭代器)再接着算
a, b, c = b, c, a+b+c
count += 1
for i in fib_bro(10):
print(i)
print('---------------------')
f = fib_unc(5)
print(f.__next__())
print(f.__next__())
print(f.__next__())
print(f.__next__())
print(f.__next__())
一个带有 yield 的函数就是一个 generator,它和普通函数不同,生成一个 generator 看起来像函数调用,但不会执行任何函数代码,直到对其调用 next()(在 for 循环中会自动调用 next())才开始执行。