python菜鳥教程學習8:迭代器與生成器
阿新 • • 發佈:2020-11-24
迭代器
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 returnx 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()