生成器里 return 的作用是什么

這篇文章主要講解了“生成器里 return 的作用是什么”，文中的講解內容簡單清晰，易于學習與理解，下面請大家跟著小編的思路慢慢深入，一起來研究和學習“生成器里 return 的作用是什么”吧！

在粉絲群里，有一位同學問了這樣一個問題：

def gen_data(num):     if num > 10:         for i in range(num):             yield i     else:         return num  generator = gen_data(5) for num in generator:     print(num)

當傳入的的參數小于等于10 的時候，為什么沒有返回這個參數本身?

這道題，當我們傳入的參數大于10的時候，能得到符合預期的結果，如下圖所示：

但是，當我們傳入數據5的時候，我們來看看運行效果：

可以看到，數字5并沒有被打印出來，程序直接運行到了最后。

之所以會出現這種情況，是因為這個同學以為，當參數大于10的時候，gen_data(12)返回的是生成器，而當參數不大于10的時候，返回的是一個數字。顯然這樣的想法是不對的，否則，for  num in 10這種語法早就報錯了，數字是不能被迭代的。

正確的說法應該是，因為gen_data里面有yield，所以gen_data(參數)返回一個生成器。無論參數傳入的是什么，返回的都是生成器。如下圖所示：

為了說明為什么傳入參數為5的時候，for 循環不執行，我們簡化一下代碼：

def gen_data():     yield 1     yield 2     yield 3     return 4  generator = gen_data() for num in generator:     print(num)

運行效果如下圖所示：

可以看到，對于這樣一個非常簡單的生成器，在 for 循環里面也只是打印了數字123，并沒有打印數字4。

關于生成器中的return，我們可以從 Python 官方文檔PEP 255 — Simple Generators[1]中找到說明：

return  在生成器中，表示生成器運行完成了，可以結束了。然后生成器會拋出一個StopIteration的異常。而for循環能夠檢測到這個異常，于是結束循環。所以當我們傳入的參數為5的時候，生成器直接運行到了  return，于是它直接就拋出StopIteration，于是 for 循環檢測到這個異常就結束了。

在生成器里面的return只是一個結束標志，它不會把后面寫的值返回給調用者。這跟函數里面的return語句是不一樣的。

感謝各位的閱讀，以上就是“生成器里 return 的作用是什么”的內容了，經過本文的學習后，相信大家對生成器里 return 的作用是什么這一問題有了更深刻的體會，具體使用情況還需要大家實踐驗證。這里是億速云，小編將為大家推送更多相關知識點的文章，歡迎關注！