迭代器

迭代是Python最強大的功能之一，是訪問元素集合的一種方法。

迭代器是一個可以記住遍歷的位置的對象。

迭代器對象從集合的第一個元素開始訪問，直到所有的元素被訪問完結束，迭代器只能向前不會后退。

迭代器有兩個基本方法，iter()和next()。

字符串，列表，或元組對象都可以用于創建迭代器。

?

?

迭代器對象可以使用常規語句for進行遍歷：

?

?

?

使用next()函數：

#next.py import syslist = [1,2,3,4] it = iter(list)while True:try:print(next(it))except StopIteration:sys.exit()

運行結果：

robot@ubuntu:~/wangqinghe/python/20190827$ python3.5 next.py

1

2

3

4

?

創建一個迭代器：

把一個類作為一個迭代器使用需要在類中實現兩個方法__iter__()與__next__().

類都是由一個構造函數，Python的構造函數為__init__(),它會在對象初始化的時候執行。

__iter__方法返回一個特殊的迭代器對象，這個迭代器對象實現了__next__()方法通過StopIteration異常標識迭代的完成。

__next__()方法會返回下一個迭代器對象。

?

#iter.py class MyNumbers:def __iter__(self):self.a = 1return selfdef __next__(self):x = self.aself.a += 1return xmyclass = MyNumbers() myiter = iter(myclass)print(next(myiter)) print(next(myiter)) print(next(myiter)) print(next(myiter)) print(next(myiter)) print(next(myiter))

運行結果：

robot@ubuntu:~/wangqinghe/python/20190827$ python3.5 iter.py

1

2

3

4

5

6

?

StopIteration

StopIteration異常用于標識迭代的完成，防止出現無限循環的情況，在__next__方法種我們可以設置在完成指定循環次數后觸發StopIteration異常來結束迭代。

?

在20次迭代后停止執行：

#stop.py class MyNumbers:def __iter__(self):self.a = 1return selfdef __next__(self):if self.a <= 20:x = self.aself.a += 1return xelse:raise StopIterationmyclass = MyNumbers() myiter = iter(myclass)for x in myiter:print(x)

運行結果：

robot@ubuntu:~/wangqinghe/python/20190827$ python3.5 stop.py

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

?

生成器：

在Python中，使用了yield的函數被稱為生成器（generator）。

跟普通函數不同，生成器是一個返回迭代器的函數，只能用于迭代操作，更簡單點理解生成器就是一個迭代器。

在調用生成器運行過程中，每次遇到yield時函數會暫停并保持當前所有運行信息，返回yield的值，并在下一次執行next()方法時從當前位置繼續運行。

調用一個生成器函數，返回的是一個迭代器對象。

#yield.py import sysdef fibonacci(n):a,b,counter = 0,1,0while True:if(counter > n):return yield aa,b = b,a+bcounter += 1 f = fibonacci(10)while True:try:print(next(f),end=" ")except StopIteration:sys.exit()

運行結果：

robot@ubuntu:~/wangqinghe/python/20190827$ python3 yield.py

0 1 1 2 3 5 8 13 21 34 55

?

什么時候需要用到yield

一個函數f，返回一個list，這個list是動態計算出來的，并且這個list會很大，這個時候我們希望每次調用這個函數并使用迭代器進行循環的時候，一個一個的得到每個list的值，而不是直接得到一個list來節省內存，這個時候yield就很有用。

?

轉載于:https://www.cnblogs.com/wanghao-boke/p/11419965.html

總結

以上是生活随笔為你收集整理的Python3迭代器和生成器的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。