快捷導(dǎo)航

Python生成器之yield詳解

更新時間：2023年05月21日 08:59:42 作者：程序員喵大人

這篇文章主要介紹了Python生成器yield，yield除了作為生成器的標(biāo)志以外，還有一個「返回值」的功能，我們知道return也有這個功能，那么它跟return的這個返回值有什么區(qū)別呢，本文將詳細(xì)的介紹yield,需要的朋友可以參考下

yield

為了搞清楚 yield 和 return 這兩者的區(qū)別，我們先來看一個簡單的例子：

>>> def self_return(n):

...    print('rocky')

...    while n > 0:

...            print('before return')

...            return n

...            n -= 1

...            print('after return')

...

>>> s = self_return(3)

rocky

before return

>>> s

3

從上面的例子中函數(shù) self_return(n) 被調(diào)用的過程中我們可以清晰的看出，s = self_return(3) 函數(shù)體內(nèi)的語句就開始執(zhí)行了，遇到 return 以后將值返回，并結(jié)束在函數(shù)體內(nèi)的執(zhí)行，所以我們看到的結(jié)果是 return 后面的語句根本沒有執(zhí)行，這個是 return 的特點(diǎn)，不知道你還記得么？如果不記得的話可以去翻我前面的文章。

下面我們來將 return 換乘 yield ，再來試試看：

>>> def self_yield(n):

...    print('rocky')

...    while n > 0:

...            print('before yield')

...            yield n

...            n -= 1

...            print('after yield')

...

>>> s = self_yield(3)

>>> s.__next__()

rocky

before yield

3

仔細(xì)觀察上面的例子你會發(fā)現(xiàn)，s = self_yield(n) 并沒有去執(zhí)行函數(shù)體內(nèi)的語句，且 s.next() 的時候遇到 yield 的時候，會返回值，并且暫停。我們接著再繼續(xù)來試一下：

>>> s.__next__()

after yield

before yield

2

>>> s.__next__()

after yield

before yield

1

>>> s.__next__()

after yield

Traceback (most recent call last):

 File "<stdin>", line 1, in <module>

StopIteration

通過上面的繼續(xù)操作，我們可以看到每次遇到 yield 的時候都會返回值，并且暫停，下次再執(zhí)行的時候是從上次暫停的位置開始又繼續(xù)執(zhí)行的，當(dāng)沒有滿足條件的值，就會拋出異常。

結(jié)合上面的分析和對用例的執(zhí)行結(jié)果，相信你已經(jīng)你已經(jīng)理解了 yield 的特點(diǎn)，也知道它與 return 之間的區(qū)別了：一般的函數(shù)，都是止于 return；作為生成器的函數(shù)，因?yàn)橛辛?yield，則遇到它會掛起。

下面我想再用一個例子來具體的闡述一下。斐波那契數(shù)列相信你們已經(jīng)不陌生了，我在前面的文章中不止一次的提過它，這次我們嘗試將 yield 應(yīng)用到斐波那契數(shù)列中：

def fibs(max):

   """

   fibonacci sequence generator

   """

   n, a, b = 0, 0, 1

   while n < max:

       yield b

       a, b = b, a + b

       n += 1



if __name__ == "__main__":

   f = fibs(9)

   for i in f:

       print(i,end = ' ')

上述代碼的運(yùn)行結(jié)果如下：