云計算開發(fā)學習筆記:Python3迭代器與生成器
迭代器
迭代是Python最強大的功能之一,是訪問集合元素的一種方式。
迭代器是一個可以記住遍歷的位置的對象。
迭代器對象從集合的第一個元素開始訪問,直到所有的元素被訪問完結(jié)束。迭代器只能往前不會后退。
迭代器有兩個基本的方法:iter() 和 next()。
字符串,列表或元組對象都可用于創(chuàng)建迭代器:
迭代器對象可以使用常規(guī)for語句進行遍歷:
執(zhí)行以上程序,輸出結(jié)果如下:
也可以使用 next() 函數(shù):
執(zhí)行以上程序,輸出結(jié)果如下:
創(chuàng)建一個迭代器
把一個類作為一個迭代器使用需要在類中實現(xiàn)兩個方法 __iter__() 與 __next__() 。
如果你已經(jīng)了解的面向?qū)ο缶幊?,就知道類都有一個構(gòu)造函數(shù),Python 的構(gòu)造函數(shù)為 __init__(), 它會在對象初始化的時候執(zhí)行。
__iter__() 方法返回一個特殊的迭代器對象, 這個迭代器對象實現(xiàn)了 __next__() 方法并通過 StopIteration 異常標識迭代的完成。
__next__() 方法(Python 2 里是 next())會返回下一個迭代器對象。
創(chuàng)建一個返回數(shù)字的迭代器,初始值為 1,逐步遞增 1:
執(zhí)行輸出結(jié)果為:
StopIteration
StopIteration 異常用于標識迭代的完成,防止出現(xiàn)無限循環(huán)的情況,在 __next__() 方法中我們可以設(shè)置在完成指定循環(huán)次數(shù)后觸發(fā) StopIteration 異常來結(jié)束迭代。
在 20 次迭代后停止執(zhí)行:
執(zhí)行輸出結(jié)果為:
生成器
在 Python 中,使用了 yield 的函數(shù)被稱為生成器(generator)。
跟普通函數(shù)不同的是,生成器是一個返回迭代器的函數(shù),只能用于迭代操作,更簡單點理解生成器就是一個迭代器。
在調(diào)用生成器運行的過程中,每次遇到 yield 時函數(shù)會暫停并保存當前所有的運行信息,返回 yield 的值, 并在下一次執(zhí)行 next() 方法時從當前位置繼續(xù)運行。
調(diào)用一個生成器函數(shù),返回的是一個迭代器對象。
以下實例使用 yield 實現(xiàn)斐波那契數(shù)列:
執(zhí)行以上程序,輸出結(jié)果如下: