python之yield用法

阅读更多

简单记录一下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。