Python yield generator

http://www.ibm.com/developerworks/cn/opensource/os-cn-python-yield/, Python yield 使用淺析

這篇說的很清楚

主要看到FP裡面的Lazy Seq概念, 所以想起這個...

 

包含yield函數就是產生器(generator)
什麼是產生器, 用於產生迭代器(iterator), 有什麼用?

在python裡面任何迭代器都可以用list comprehension, 當list來處理

但是對於很大的list和無限list, 你肯定不能事先把list產生出來, 所以我只想在用的時候再去產生...Lazy Sequence的概念

那麼yield就是用來方便定義這樣的迭代器, 看個Fibonacci的例子.

這個邏輯很簡單, 如果你想要這個list, 可以將b append到list裡面返回. 問題是這個不是lazy的, 必須執行完才能得到完整的list.
如果list很大, memory可能是issue, 有時也許我只需要前幾個...總之就是不靈活

def fab(max):     n, a, b = 0, 0, 1     while n < max:         print b         a, b = b, a + b         n = n + 1 

 

為了靈活, 我們產生個迭代器,這樣使用起來就很方便, 可以用for, 也可以手動的next, 需要幾個取幾個

問題是, 這個代碼實現和上面比很複雜, 可以用上面簡單的code實現迭代器嗎?

class Fab(object):     def __init__(self, max):         self.max = max         self.n, self.a, self.b = 0, 0, 1     def __iter__(self):         return self     def next(self):         if self.n < self.max:             r = self.b             self.a, self.b = self.b, self.a + self.b             self.n = self.n + 1             return r         raise StopIteration() 

>>> for n in Fab(5):  ...     print n  ...  1  1  2  3  5 

ok, yield登場, 只需要把print換成yield就實現了, 是不是很easy

def fab(max):     n, a, b = 0, 0, 1     while n < max:         yield b         # print b         a, b = b, a + b         n = n + 1 

>>> f = fab(5)  >>> f.next()  1  >>> f.next()  1  >>> f.next()  2 

這個可以用於任何lazy sequence的情境, list和內容非常多, 我想一部分一部分取, 比如讀個很大的檔案, 或其他的IO, 都可以使用.