python迭代器以及產生器__python

迭代，即重複做一些事很多次，python中通過for迴圈可以對序列、字典以及其他的對象進行迭代。在其他對象進行迭代時，需要在其他對象中實現__iter__方法。

__iter__方法返回一個迭代器，所謂的迭代器就是具有next方法的對象。在調用next方法時，迭代器會返回它的下一個值。如果next方法被調用，但迭代器沒有值可以返回，就會引發一個StopIteration異常。

用迭代器實現一個斐波那契數列

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 += 1            return r        raise StopIterationprint '迭代器的輸出結果：'for i in Fab(5):    print i

輸出結果：

除了能夠在迭代器和可迭代對象上進行迭代(這是經常做的)外，還能把它們轉換為序列。在大部分能夠使用序列的情況下，能使用迭代器替換。

產生器，是一種用普通的函數文法定義的迭代器，是包含yield語句的函數。除了名字不同以外，它的行為和普通函數也有很大的差別。這在於它不像return那樣產生傳回值，而是每次產生多個值。每次產生一個值（使用yield語句），函數機會被凍結：即函數停在那點等待被啟用。函數被啟用後就從停止的那點開始執行。函數的本地變數和上一次啟動並執行狀態一樣。

用產生器實現斐波那契數列。

def fab(max):    n, a, b = 0, 0, 1    while n < max:        yield b        a, b = b, a + b        n = n + 1print '產生器產生的結果：'for i in fab(5):    print i

用產生器解決n皇后問題。

def config(state, nextX):    nextY = len(state)    for i in range(nextY):        if abs(nextX - state[i]) in (0,nextY - i):            return True    return Falsedef nQueens(num = 8, state = ()):    if len(state) == num - 1:        for pos in range(num):            if not config(state, pos):                yield (pos)    else:        for pos in range(num):            if not config(state, pos):                for result in nQueens(num, state + (pos,)):                    yield (pos,) + resultdef prettyprint(solution):    def line(pos, lenth = len(solution)):        return '. '*(pos) + 'X ' + '. '*(lenth - pos -1)    for pos in solution:        print line(pos)i = 1for solution in nQueens():    print '第%d种放置情況' % (i)    prettyprint(solution)    i += 1