最近遇到好多python代码中使用yield的 code,表示的是一个函数是一个生成器。什么意思呢
简答就是说,如果一个函数中使用了yield 定义了一个语句,那么这个函数就不在是一个普通的函数,而是一个生成器,也就是说这个函数是一个可迭代对象。
看下面一段代码:
def yield_test(n):
for i in range(n):
yield callMe(i) #a
print("i=",i) #b
print("something else")
print("end.")
def callMe(i):
return i*i
for i in yield_test(3):
print(i)
当第一次调用 函数 yield_test ,定义的函数中的code不会像调用其他方法那样依次执行完整个方法,而是在执行到 语句yield (place a 的地方)时,返回后面的函数调用结果。
然后在 for 执行第二次的时候,函数直接从 b 位置开始执行,遇到 yield 语句后,在返回到 for 语句中。依次把for 语句执行完。
试行结果如下:
0
i= 0
1
i= 1
4
i= 2
something else
end.
也可以直接以可迭代对象的方式调用 生成器:yield_test(3).yield_test(3)._next_() (python3.x 以前为next() 函数), 每调用一次返回一个值。
使用生成器的好处是:
- 生成器会把函数变成一个可迭代对象,每次返回一个值给caller,减少内存消耗(同样的问题不用生成器,可能需要使用list 储存数据然后返回给 caller)
- 在第二次迭代生成器时,yield 语句之前的变量值会被保存,像是函数被挂住了,第二次会继续执行余下代码,而不是从函数第一行重新开始。