循環器(iterator)的功能
循環器(迭代器)是對象的容器,包含有多個對象。通過調用循環器的next()方法 (__next__()方法,在Python 3.x中),循環器將依次返回一個對象。直到所有的對象遍歷窮盡,循環器將舉出StopIteration錯誤。
在for i in iterator結構中,循環器每次返回的對象將賦予給i,直到循環結束。使用iter()內置函數,我們可以將諸如表、字典等容器變爲循環器。比如
for i in iter([2, 4, 5, 6]):
print(i)
>>> a=iter([2, 4, 5, 6])
>>> while True:
next(a)
標準庫中的itertools包提供了更加靈活的生成循環器的工具。這些工具的輸入大都是已有的循環器。另一方面,這些工具完全可以自行使用Python實現,該包只是提供了一種比較標準、高效的實現方式。這也符合Python“只有且最好只有解決方案”的理念。
# import the tools
from itertools import *
無窮循環器
count(5, 2) #從5開始的整數循環器,每次增加2,即5, 7, 9, 11, 13, 15 ...
cycle('abc') #重複序列的元素,既a, b, c, a, b, c ...
repeat(1.2) #重複1.2,構成無窮循環器,即1.2, 1.2, 1.2, ...
repeat也可以有一個次數限制:
repeat(10, 5) #重複10,共重複5次
>>> a=cycle('abc')
>>> a
<itertools.cycle object at 0x0000000003141D48>
>>> next(a)
'a'
>>> next(a)
'b'
>>>
>>> next(a)
'c'