迭代器
迭代式訪問元素的一種方式,迭代器是一個可以記住遍歷的位置的對象。
迭代器對象從集合的第一個元素開始訪問,直到所有的元素被訪問完結束。迭代器只能往前不會后退。
迭代器有兩個基本的方法: iter() ?和? next() 。
常見的字符串,列表或元組對象都可用于創(chuàng)建迭代器:
>>> list=[1,2,3,4,5
]
>>> it=
iter(list)
>>>
print
(next(it))
1
>>>
print
(next(it))
2
迭代器對象可以使用常規(guī)for語句進行遍歷:
list=[1,2,3,4,5
]
it
= iter(list)
#
創(chuàng)建迭代器對象
for
x
in
it:
print
(x, end=
"
"
)
輸出如下:
1 2 3 4 5
也可以使用 next() 函數:
import
sys
#
引入 sys 模塊
list
= [1, 2, 3, 4
]
it
= iter(list)
#
創(chuàng)建迭代器對象
while
True:
try
:
print
(next(it))
except
StopIteration:
sys.exit()
輸出如下:
1
2
3
4
創(chuàng)建一個迭代器
把一個類作為一個迭代器使用需要在類中實現兩個方法 __iter__() 與 __next__() 。
如果你已經了解的面向對象編程,就知道類都有一個構造函數,Python 的構造函數為 __init__(), 它會在對象初始化的時候執(zhí)行。
__iter__() 方法返回一個特殊的迭代器對象, 這個迭代器對象實現了 __next__() 方法并通過 StopIteration 異常標識迭代的完成。
__next__() 方法(Python 2 里是 next())會返回下一個迭代器對象。
創(chuàng)建一個返回數字的迭代器,初始值為 1,逐步遞增 1:
class
MyNum:
def
__iter__
(self):
self.a
=1
return
self
def
__next__
(self):
x
=
self.a
self.a
+= 1
return
x
test
=
MyNum()
myiter
=
iter(test)
print
(next(myiter))
print
(next(myiter))
print
(next(myiter))
print
(next(myiter))
輸出如下:
1
2
3
4
StopIteration
StopIteration 異常用于標識迭代的完成,防止出現無限循環(huán)的情況,在 __next__() 方法中我們可以設置在完成指定循環(huán)次數后觸發(fā) StopIteration 異常來結束迭代。
在 10 次迭代后停止執(zhí)行:
class
MyNum:
def
__iter__
(self):
self.a
=1
return
self
def
__next__
(self):
if
self.a<=10
:
x
=
self.a
self.a
+= 1
return
x
else
:
raise
StopIteration
test
=
MyNum()
myiter
=
iter(test)
for
i
in
myiter:
print
(i)
輸出結果:
1
2
3
4
5
6
7
8
9
10
生成器
在 Python 中,使用了 yield 的函數被稱為生成器(generator)。
跟普通函數不同的是,生成器是一個返回迭代器的函數,只能用于迭代操作,更簡單點理解生成器就是一個迭代器。
在調用生成器運行的過程中,每次遇到 yield 時函數會暫停并保存當前所有的運行信息,返回 yield 的值, 并在下一次執(zhí)行 next() 方法時從當前位置繼續(xù)運行。
調用一個生成器函數,返回的是一個迭代器對象。
以下源碼實現使用yield實現斐波那契數列:
def
myfib(n):
#
生成器函數 - 斐波那契
a,b,counter = 0,1
,0
while
True:
if
counter>
n:
return
yield
a
a,b
= b,a+
b
counter
+= 1
f
=myfib(10)
#
f 是一個迭代器,由生成器返回生成
while
True:
try
:
print
(next(f))
except
StopIteration:
break
輸出結果:
0
1
1
2
3
5
8
13
21
34
55
?
更多文章、技術交流、商務合作、聯(lián)系博主
微信掃碼或搜索:z360901061
微信掃一掃加我為好友
QQ號聯(lián)系: 360901061
您的支持是博主寫作最大的動力,如果您喜歡我的文章,感覺我的文章對您有幫助,請用微信掃描下面二維碼支持博主2元、5元、10元、20元等您想捐的金額吧,狠狠點擊下面給點支持吧,站長非常感激您!手機微信長按不能支付解決辦法:請將微信支付二維碼保存到相冊,切換到微信,然后點擊微信右上角掃一掃功能,選擇支付二維碼完成支付。
【本文對您有幫助就好】元

