Python 迭代器
Python迭代器
迭代器是一個包含可計數值的物件。
迭代器是一個可以進行迭代的物件,這意味著您可以遍歷所有值。
嚴格來說,在 Python 中,迭代器是實現了迭代器協議的物件,該協議包含 __iter__()
和 __next__()
方法。
迭代器 vs 可迭代物件
列表、元組、字典和集合都是可迭代物件。它們是可迭代的容器,您可以從中獲取迭代器。
所有這些物件都有一個 iter()
方法,用於獲取迭代器
示例
從元組返回迭代器,並列印每個值
mytuple = ("apple", "banana", "cherry")
myit = iter(mytuple)
print(next(myit))
print(next(myit))
print(next(myit))
自己動手試一試 »
即使是字串也是可迭代物件,並且可以返回迭代器
示例
字串也是可迭代物件,包含一系列字元
mystr = "banana"
myit = iter(mystr)
print(next(myit))
print(next(myit))
print(next(myit))
print(next(myit))
print(next(myit))
print(next(myit))
自己動手試一試 »
遍歷迭代器
我們也可以使用 for
迴圈來遍歷可迭代物件
實際上,for
迴圈會建立一個迭代器物件,並在每次迴圈時執行 next() 方法。
建立迭代器
要將物件/類建立為迭代器,您必須為您的物件實現 __iter__()
和 __next__()
方法。
正如您在 Python 類/物件章節中學到的,所有類都有一個名為 __init__()
的函式,它允許您在建立物件時進行一些初始化。
__iter__()
方法的作用類似,您可以執行操作(初始化等),但必須始終返回迭代器物件本身。
__next__()
方法也允許您執行操作,並且必須返回序列中的下一個項。
示例
建立一個返回數字的迭代器,從 1 開始,每個序列遞增一(返回 1,2,3,4,5 等)
class MyNumbers
def __iter__(self)
self.a = 1
return self
def __next__(self)
x = self.a
self.a += 1
return x
myclass = MyNumbers()
myiter = iter(myclass)
print(next(myiter))
print(next(myiter))
print(next(myiter))
print(next(myiter))
print(next(myiter))
自己動手試一試 »
StopIteration
如果您有足夠的 next() 語句,或者在 for
迴圈中使用,上面的示例將永遠繼續下去。
為了防止迭代無限進行,我們可以使用 StopIteration
語句。
在 __next__()
方法中,我們可以新增一個終止條件,在迭代次數達到指定次數時引發錯誤
示例
迭代 20 次後停止
class MyNumbers
def __iter__(self)
self.a = 1
return self
def __next__(self)
if self.a <= 20
x = self.a
self.a += 1
return x
else
raise StopIteration
myclass = MyNumbers()
myiter = iter(myclass)
for x in myiter
print(x)
自己動手試一試 »