python之yield用法

hotnet

　　简单记录一下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返回迭代值。供[size=1em]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。