1. 程式人生 > 實用技巧 >python菜鳥教程學習8:迭代器與生成器

python菜鳥教程學習8:迭代器與生成器

迭代器

  python最強大的功能之一,訪問集合元素的一種方式。迭代器是一個可以記住遍歷的位置的物件,迭代器物件從集合的第一個元素開始訪問,直到所有的元素被訪問完結束,迭代器只能往前不會後退。

  迭代器有兩個基本的方法:

  • iter()
  • next()

  字串,列表和元組物件都可用於建立迭代器

>>> list=[1,2,3,4]
>>> it = iter(list)    # 建立迭代器物件
>>> print (next(it))   # 輸出迭代器的下一個元素

建立一個迭代器

  把一個類作為一個迭代器使用需要在類中實現兩個方法 __iter__() 與 __next__() 。__iter__() 方法返回一個特殊的迭代器物件, 這個迭代器物件實現了 __next__() 方法並通過 StopIteration 異常標識迭代的完成。__next__() 方法會返回下一個迭代器物件。

StopIteration

  StopIteration 異常用於標識迭代的完成,防止出現無限迴圈的情況,在 __next__() 方法中我們可以設定在完成指定迴圈次數後觸發 StopIteration 異常來結束迭代。

 1 class MyNumbers:
 2   def __iter__(self):
 3     self.a = 1
 4     return self
 5  
 6   def __next__(self):
 7     if self.a <= 20:
 8       x = self.a
 9       self.a += 1
10       return
x 11 else: 12 raise StopIteration 13 14 myclass = MyNumbers() 15 myiter = iter(myclass) 16 17 for x in myiter: 18 print(x)
  • 由這個程式碼可知for迴圈可以約等於不斷呼叫next()
  • next()函式內部的x是為了記錄返回一個初始值

生成器

  在 Python 中,使用了 yield 的函式被稱為生成器(generator)。跟普通函式不同的是,生成器是一個返回迭代器的函式,只能用於迭代操作,更簡單點理解生成器就是一個迭代器。

  在呼叫生成器執行的過程中,每次遇到 yield 時函式會暫停並儲存當前所有的執行資訊,返回 yield 的值, 並在下一次執行 next() 方法時從當前位置繼續執行。呼叫一個生成器函式,返回的是一個迭代器物件。

 1 import sys
 2  
 3 def fibonacci(n): # 生成器函式 - 斐波那契
 4     a, b, counter = 0, 1, 0
 5     while True:
 6         if (counter > n): 
 7             return
 8         yield a
 9         a, b = b, a + b
10         counter += 1
11 f = fibonacci(10) # f 是一個迭代器,由生成器返回生成
12  
13 while True:
14     try:
15         print (next(f), end=" ")
16     except StopIteration:
17         sys.exit()