迭代器
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()
轉載請註明出處,本文鏈接:https://www.uj5u.com/houduan/227071.html
標籤:其他
上一篇:Java基礎學習路線配套視頻
