生成器和迭代器

如果列表元素可以按照某種演算法推算出來，那我們是否可以在迴圈的過程中不斷推算出後續的元素呢？這樣就不必建立完整的list，從而節省大量的空間。在python中，這種一邊迴圈一邊計算的機制，稱為生成器：generator。

要建立乙個generator，有很多種方法。第一種方法很簡單，只要把乙個列表生成式的改成()，就建立了乙個generator：

>>> g = (x * x for x in range(10))
>>>g
at 0x1022ef630>

我們可以直接列印出list的每乙個元素，但我們怎麼列印出generator的每乙個元素呢？

最難理解的就是generator和函式的執行流程不一樣。函式是順序執行，遇到return語句或者最後一行函式語句就返回。而變成generator的函式，在每次呼叫next()的時候執行，遇到yield語句返回，再次執行時從上次返回的yield語句處繼續執行。

def
fib(max):
n, a, b = 0, 0, 1
while n yield
b        a, b = (b, a +b)
n += 1
for i in fib(6):
print(i)

我們在迴圈過程中不斷呼叫yield，就會不斷中斷。當然要給迴圈設定乙個條件來退出迴圈，不然就會產生乙個無限數列出來。

>>> g = fib(6)
>>> while
true:
...     
try:
...         x =next(g)
...         
print('g:'
, x)
...     
except
stopiteration as e:
...         
print('
generator return value:
', e.value)
...         
break
...g: 1g: 1g: 2g: 3g: 5g: 8generator 
return value: done

把list、dict、str等iterable變成iterator可以使用iter()函式：

生成器和迭代器

生成器和迭代器

迭代器和生成器

生成器和迭代器

相關推薦