快捷導(dǎo)航

Python3?中return和yield的區(qū)別

更新時間：2022年06月07日 15:35:17 作者：小小程序員ol

這篇文章主要介紹了Python3?中return和yield的區(qū)別，return和yield都用來返回值；在一次性地返回所有值場景中return和yield的作用是一樣的，但是具體有什么區(qū)別呢，帶著疑問一起進(jìn)入下面文章學(xué)習(xí)詳細(xì)內(nèi)容吧

一、前言

return一直中，每中語言中其沒沒有很大差別，就不多說了。（shell語言return的是退出狀態(tài)，可能差別是比較大的）最早看到y(tǒng)ield應(yīng)該是哪們語言用來調(diào)整什么線程優(yōu)先級的，記不清了，不過那里的yield和python中的yield應(yīng)該功能有區(qū)別。

python中最早看到y(tǒng)ield應(yīng)該是使用scrapy框架寫爬蟲的時候，之前也有去看yiled的用法，總記不太住。今天又去看了一下，基本上來就是講些斐波那契數(shù)列的煩的要死，自己寫段程序研究了一下，這里記一下。

二、return和yield的異同

共同點：return和yield都用來返回值；在一次性地返回所有值場景中return和yield的作用是一樣的。

不同點：如果要返回的數(shù)據(jù)是通過for等循環(huán)生成的迭代器類型數(shù)據(jù)（如列表、元組），return只能在循環(huán)外部一次性地返回，yeild則可以在循環(huán)內(nèi)部逐個元素返回。下邊我們舉例說明這個不同點。

三、實例說明

1 return版本

示例代碼如下：

class TestYield:
    def gen_iterator(self):
        result_list = []
        for j in range(3):
            print(f"gen_iterator-{j}")
            result_list.append(j)
        # return在循環(huán)的外部，待變量完全生成后一次性返回
        return result_list

    def call_gen_iterator(self):
        # 執(zhí)行下邊這句后result_list直接是完成的結(jié)果[0,1,2]
        result_list = self.gen_iterator()
        for i in result_list:
            print(f"call_gen_iterator-{i}")

if __name__ == "__main__":
    obj = TestYield()
    obj.call_gen_iterator()

執(zhí)行結(jié)果如下，可以看到一次性執(zhí)行完下層函數(shù)，生成完整的迭代器類型返回值result_list，一次性返回給上層函數(shù)：

2 yield版本

示例代碼如下：

class TestYield:
    def gen_iterator(self):
        for j in range(3):
            print(f"do_something-{j}")
            # yield在for循環(huán)內(nèi)部
            yield j

    def call_gen_iterator(self):
        # yield并不是直接返回[0,1,2]，執(zhí)行下邊這句后result_list什么值都沒有
        result_list = self.gen_iterator()
        # i每請求一個數(shù)據(jù)，才會觸發(fā)gen_iterator生成一個數(shù)據(jù)
        for i in result_list:
            print(f"call_gen_iterator-{i}")

if __name__ == "__main__":
    obj = TestYield()
    obj.call_gen_iterator()

執(zhí)行結(jié)果如下，可以看到上下層函數(shù)是交替進(jìn)行的，即上層函數(shù)請求迭代一個值下層函數(shù)才生成一個值并立即返回這個值：

3 yield的意義

從上邊兩個小節(jié)可以看到，雖然return和yield兩者執(zhí)行的順序有區(qū)別，但整個要做的事情是一樣的，所以使用yield并不會比return快，甚至我們可以猜測由于yield總發(fā)生上下文切換在速度上還會慢一些，所以速度不是yield的意義。

他們的主要區(qū)別是yiled要迭代到哪個元素那個元素才即時地生成，而return要用一個中間變量result_list保存返回值，當(dāng)result_list的長度很長且每個組成元素內(nèi)容很大時將會耗費比較大的內(nèi)存，此時yield相對return才有優(yōu)勢。

四、yield和return嵌套使用

class TestYield:
    def gen_iterator(self):
        for j in range(3):
            print(f"do_something-{j}")
            # yield在for循環(huán)內(nèi)部
            yield j

    def gen_iterator_middle(self):
        print(f"gen_iterator_middle")
        # 返回的是迭代器的句柄，所以加一層return不影響是可以理解的
        return self.gen_iterator()

    def call_gen_iterator(self):
        # yield并不是直接返回[0,1,2]，執(zhí)行下邊這句后result_list什么值都沒有
        result_list = self.gen_iterator_middle()
        # i每請求一個數(shù)據(jù)，才會觸發(fā)gen_iterator生成一個數(shù)據(jù)
        for i in result_list:
            print(f"call_gen_iterator-{i}")

if __name__ == "__main__":
    obj = TestYield()
    obj.call_gen_iterator()