前言

在Python中存在兩種好用的功能迭代器與生成器。當(dāng)我們剛開(kāi)始接觸到時(shí)，會(huì)簡(jiǎn)單的認(rèn)為迭代只不過(guò)是處理列表、集合等序列中元素而已。然而，它們的功能并非如此。

當(dāng)我們遇到一個(gè)列表[1,2,3,4,5,6]需要進(jìn)行遍歷時(shí)，我們可以使用如下方法：

• for循環(huán)遍歷打印
• next()方法打印

同時(shí)，我們也可以創(chuàng)建個(gè)性化的迭代器對(duì)象、構(gòu)造生成器函數(shù)等等。

Python中迭代器協(xié)議主要用到了兩個(gè)魔法方法：__iter__(),__next__()

• __iter__() 方法創(chuàng)建一個(gè)具有__next__()方法的迭代器對(duì)象
• __next__() 方法返回下一個(gè)迭代器對(duì)象

Python中只具有迭代操作的生成器，也是屬于迭代器的。在Python中，我們可以使用yield的函數(shù)來(lái)實(shí)現(xiàn)生成器。

本期，我們將詳細(xì)介紹生成器相關(guān)原理和用法，Let's go~~~

1. 什么是yield?

首先我們需要認(rèn)識(shí)一生成器（generator），從字面意思上理解，循環(huán)計(jì)算的操作方式。在Python中，提供一種可以邊循環(huán)邊計(jì)算的機(jī)制。

生成器是解決使用序列存儲(chǔ)大量數(shù)據(jù)時(shí)，內(nèi)存消耗大的問(wèn)題。我們可以根據(jù)存儲(chǔ)數(shù)據(jù)的某些規(guī)律，演算為算法，在循環(huán)過(guò)程中通過(guò)計(jì)算得到，這樣可以不用創(chuàng)建完整序列，從而大大節(jié)省占用空間。

yield 是實(shí)現(xiàn)生成器方法之一，當(dāng)函數(shù)使用yield方法，則該函數(shù)就成為了一個(gè)生成器。調(diào)用該函數(shù)，就等于創(chuàng)建了一個(gè)生成器對(duì)象。

2. yield 原理

一個(gè)生成器，主要是通過(guò)循環(huán)反復(fù)調(diào)用next()方法，直到捕獲異常。

具有 yield方法的函數(shù)也是一個(gè)生成器，創(chuàng)建如下栗子：

def test(n):
    print("starting...")
    while True:
        res = yield n
        print ("res:",res)
g = test(5)
print(next(g))
print("######")
print(next(g))

我們可以看到輸出的打印日志：

• test函數(shù)帶有yield關(guān)鍵字，說(shuō)明它是一個(gè)生成器，不會(huì)進(jìn)行執(zhí)行
• 當(dāng)test函數(shù)遇到next()方法時(shí)，開(kāi)始執(zhí)行test函數(shù)內(nèi)部步驟
• 直到程序遇到y(tǒng)ield關(guān)鍵字時(shí)，程序會(huì)中止
• 直到下一次的next()方法喚醒，執(zhí)行yield后續(xù)步驟，print()方法。這時(shí)候，res沒(méi)有被賦值。
• 再次進(jìn)入循環(huán)while內(nèi)部步驟，同理遇到y(tǒng)ield關(guān)鍵字中止循環(huán)

當(dāng)我們需要對(duì)生成器里的元素進(jìn)行賦值時(shí)，我們可以調(diào)用.send()方法：

因此，帶yield的函數(shù)具體內(nèi)部執(zhí)行操作為：

• yield 方法：相當(dāng)于Return作用，程序遇到y(tǒng)ield則直接中止后續(xù)步驟
• 當(dāng)再次調(diào)用生成器時(shí)，next()方法會(huì)喚醒，并繼續(xù)執(zhí)行yield后續(xù)步驟
• 還可以調(diào)用send()方法，可以喚醒，并傳入一個(gè)值，繼續(xù)執(zhí)行yield后續(xù)步驟

生成器是可迭代的，每一次只可讀一次。因此常常與for循環(huán)一起組合使用。

3. yield 常見(jiàn)用法

我們通?？梢允褂脦в衴ield的函數(shù)來(lái)創(chuàng)建生成器來(lái)替代包含大量數(shù)據(jù)的序列。

import sys
def test(n):
    print("start")
    while n > 0:
        yield n
        n-=1
    print("end")
a = [x for x in range(1000)]
b= test(1000)
print("a內(nèi)存大?。?,sys.getsizeof(a))
print("b內(nèi)存大?。?,sys.getsizeof(b))

總結(jié)

本期，我們主要對(duì)生成器及關(guān)鍵字yield的相關(guān)細(xì)節(jié)點(diǎn)進(jìn)行學(xué)習(xí)。生成器是迭代器的一中，只用來(lái)迭代操作。生成器內(nèi)部主要是調(diào)用next()方法或send()方法來(lái)訪問(wèn)下一個(gè)迭代器對(duì)象。我們可以定義帶有yield關(guān)鍵字的函數(shù)來(lái)實(shí)現(xiàn)生成器，yield相當(dāng)于return作用，但是可以支持傳參。

到此這篇關(guān)于Python 生成器yield原理及用法的文章就介紹到這了,更多相關(guān)Python 生成器yield內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論