迭代器與生成器

迭代器與生成器

1.1 迭代器

• 字符串，列表或元組對象都可用於創建迭代器：
• 迭代器是一個可以記住遍歷的位置的對象。
• 迭代器對象從集合的第一個元素開始訪問，直到所有的元素被訪問完結束。迭代器只能往前不會後退。
• 迭代器有兩個基本的方法：iter() 和 next()。

list=[1,2,3,4,5,6,7,8,9]
itr = iter(list)
print(next(itr))

1

print(next(itr))

2

使用for循環遍歷迭代器

迭代不會向後，之前使用next方法取出的元素不會再被遍歷。

for i in itr:
    print(i)

3
4
5
6
7
8
9

創建一個迭代器

需要實現 __iter__() 與 __next__() 方法

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))

1
2
3
4
5

2.1 生成器

使用了 yield 的函數被稱爲生成器. 生成器是一個返回迭代器的函數

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

import sys
 
def fibonacci(n): # 生成器函數 - 斐波那契
    a, b, counter = 0, 1, 0
    while True:
        if (counter > n): 
            return
        yield a
        a, b = b, a + b
        counter += 1
f = fibonacci(10) # f 是一個迭代器，由生成器返回生成
 
while True:
    try:
        print (next(f), end=" ")
    except StopIteration:
        sys.exit()

0 1 1 2 3 5 8 13 21 34 55

An exception has occurred, use %tb to see the full traceback.

SystemExit

F:\software\anaconda\envs\Tensorflow1\lib\site-packages\IPython\core\interactiveshell.py:2870: UserWarning: To exit: use ‘exit’, ‘quit’, or Ctrl-D.
warn(“To exit: use ‘exit’, ‘quit’, or Ctrl-D.”, stacklevel=1)

• 關於yeild的其他介紹：

一個帶有 yield 的函數就是一個 generator，它和普通函數不同，生成一個 generator 看起來像函數調用，但不會執行任何函數代碼，直到對其調用 next()（在 for 循環中會自動調用 next()）纔開始執行。雖然執行流程仍按函數的流程執行，但每執行到一個 yield 語句就會中斷，並返回一個迭代值，下次執行時從 yield 的下一個語句繼續執行。看起來就好像一個函數在正常執行的過程中被 yield 中斷了數次，每次中斷都會通過 yield 返回當前的迭代值。