使用生成器創(chuàng)建新的迭代模式

問題

你想實現(xiàn)一個自定義迭代模式，跟普通的內(nèi)置函數(shù)比如 range() , reversed() 不一樣。

解法

如果你想實現(xiàn)一種新的迭代模式，使用一個生成器函數(shù)來定義它。下面是一個生產(chǎn)某個范圍內(nèi)浮點數(shù)的生成器：

def frange(start, stop, increment):x = startwhile x < stop:yield xx += increment

為了使用這個函數(shù)，你可以用 for 循環(huán)迭代它或者使用其他接受一個可迭代對象的函數(shù) (比如 sum() , list() 等)。示例如下：

>>> for n in frange(0, 4, 0.5): ... print(n) ... 0 0.5 1.0 1.5 2.0 2.5 3.0 3.5 >>> list(frange(0, 1, 0.125)) [0, 0.125, 0.25, 0.375, 0.5, 0.625, 0.75, 0.875] >>>

討論

一個函數(shù)中需要有一個 yield 語句即可將其轉(zhuǎn)換為一個生成器。跟普通函數(shù)不同的是，生成器只能用于迭代操作。下面是一個實驗，向你展示這樣的函數(shù)底層工作機制：

>>> def countdown(n): ... print('Starting to count from', n) ... while n > 0: ... yield n ... n -= 1 ... print('Done!') ... >>> # Create the generator, notice no output appears >>> c = countdown(3) >>> c <generator object countdown at 0x1006a0af0>>>> # Run to first yield and emit a value >>> next(c) Starting to count from 3 3>>> # Run to the next yield >>> next(c) 2 >>> # Run to next yield >>> next(c) 1 >>> # Run to next yield (iteration stops) >>> next(c) Done! Traceback (most recent call last): File "<stdin>", line 1, in <module> StopIteration >>>

一個生成器函數(shù)主要特征是它只會回應(yīng)在迭代中使用到的 next 操作。一旦生成器函數(shù)返回退出，迭代終止。我們在迭代中通常使用的 for 語句會自動處理這些細節(jié)，所以你無需擔心。

總結(jié)

以上是生活随笔為你收集整理的《Python Cookbook 3rd》笔记（4.3）：使用生成器创建新的迭代模式的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。