本文轉(zhuǎn)載自微信公眾號「未聞Code」，作者kingname 。轉(zhuǎn)載本文請聯(lián)系未聞Code公眾號。

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

 
 
 
 
  
  
  
  
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) 
 
 
 

當(dāng)傳入的的參數(shù)小于等于10 的時候，為什么沒有返回這個參數(shù)本身?

這道題，當(dāng)我們傳入的參數(shù)大于10的時候，能得到符合預(yù)期的結(jié)果，如下圖所示：

但是，當(dāng)我們傳入數(shù)據(jù)5的時候，我們來看看運行效果：

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

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

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

為了說明為什么傳入?yún)?shù)為5的時候，for 循環(huán)不執(zhí)行，我們簡化一下代碼：

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

運行效果如下圖所示：

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

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

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

在生成器里面的return只是一個結(jié)束標(biāo)志，它不會把后面寫的值返回給調(diào)用者。這跟函數(shù)里面的return語句是不一樣的。

[1]PEP 255 — Simple Generators: https://www.python.org/dev/peps/pep-0255/#specification-return

名稱欄目：一日一技：生成器里面的 return 有什么用？
網(wǎng)站鏈接：http://uogjgqi.cn/article/coghhpg.html