簡(jiǎn)單介紹 yield

本篇文章會(huì)先向你介紹一個(gè)陌生的 Python 關(guān)鍵詞，他和 return 就像一對(duì)新兄弟，有相似之處，又各有不同。

相似的是：yield 和 return 都可以在一個(gè)函數(shù)里將值返回給調(diào)用方；

不同的是：return 后，函數(shù)運(yùn)行就終止了，而 yield 則只是暫停運(yùn)行。

關(guān)于 yield 的簡(jiǎn)單使用，請(qǐng)先看如下例子

>>> def demo():
...     yield 1
...     yield 2
...     yield 3
...
>>> gen = demo()
>>> gen

>>>
>>>
>>> for i in gen:
...     print(i)
...
1
2
3

重點(diǎn)有如下幾個(gè)：

含有 yield 的函數(shù)，不再是普通的函數(shù)，直接調(diào)用含有 yield 的函數(shù)，返回的是一個(gè)生成器對(duì)象（generator object）

可以使用 for 循環(huán)（實(shí)際還可以使用 list 或者 next 函數(shù)）來遍歷該生成器對(duì)象，將 yield 的內(nèi)容一個(gè)一個(gè)打印出來

向生成器中發(fā)送消息

函數(shù)暫停之后，如果調(diào)用者能在下一次恢復(fù)函數(shù)運(yùn)行的時(shí)候，向它傳遞一些信息，那么整個(gè)程序的靈活性會(huì)大大提升。

下面通過一個(gè)簡(jiǎn)明的演示來看看，如何往生成器中發(fā)送消息：

def jumping_range(N):
    index = 0
    while index < N:
        # 通過send()發(fā)送的信息將賦值給jump
        jump = yield index
        if jump is None:
            jump = 1
        index += jump

if __name__ == '__main__':
    itr = jumping_range(5)
    print(next(itr))
    print(itr.send(2))
    print(next(itr))
    print(itr.send(-1))

輸出。

0
2
3
2

這里解釋下為什么這么輸出。 重點(diǎn)是jump = yield index這個(gè)語句。

分成兩部分：

1.yield index 是將index return給外部調(diào)用程序。

2.jump = yield 可以接收外部程序通過send()發(fā)送的信息，并賦值給jump


審核編輯：湯梓紅