简单记录一下yield的用法,不说一堆的文字废话,直接从代码入手,下面的代码是摘自web.py框架中的代码
def group(seq, size):
"""
Returns an iterator over a series of lists of length size from iterable.
>>> list(group([1,2,3,4], 2))
[[1, 2], [3, 4]]
>>> list(group([1,2,3,4,5], 2))
[[1, 2], [3, 4], [5]]
"""
def take(seq, n):
for i in xrange(n):
yield seq.next()
if not hasattr(seq, 'next'):
seq = iter(seq)
while True:
x = list(take(seq, size))
if x:
yield x
else:
break
web.py中很多这样风格的代码,十分的简洁精炼,代码从上往下执行,声明了一个take函数,然后进入if语句,刚开始seq只是一个列表,所以hasattr(seq,'next')为None,所以if not None成立,进入了seq=iter(seq),那么seq就变成了一个iter对象,然后进入while True循环,进行调用了take方法,其实这里虽然说是调用,但是实际上不会去执行take里面的代码,这个方法只是简单的返回了一个iterable 对象,list(iterable)会类似for循环来执行 fab 函数内部的代码并且返回iter.next()值,执行到 yield seq.next()时,take函数就返回一个迭代值,下次迭代时,代码从 yield seq.next()的下一条语句继续执行,所以list(iterable)将返回一个列表,if x成立,那么yield x返回迭代值。供list(group([1,2,3,4], 2))中的list进行迭代。
例如下面一个例子
>>> for n in fab(5):
... print n
...
1
1
2
3
5
def fab(max):
n, a, b = 0, 0, 1
while n < max:
yield b
# print b
a, b = b, a + b
n = n + 1
简单地讲,yield 的作用就是把一个函数变成一个 generator,带有 yield 的函数不再是一个普通函数,Python 解释器会将其视为一个 generator,调用 fab(5) 不会执行 fab 函数,而是返回一个 iterable 对象!在 for 循环执行时,每次循环都会执行 fab 函数内部的代码,执行到 yield b 时,fab 函数就返回一个迭代值,下次迭代时,代码从 yield b 的下一条语句继续执行,而函数的本地变量看起来和上次中断执行前是完全一样的,于是函数继续执行,直到再次遇到 yield。