我們一起看兩個生成器相關的練習，主要是練習一下 yield 的用法。

問題 1

輸入一個數字 n，寫一個程序判斷 0 到 n 之間的偶數，并用逗號分隔來輸出。

比如輸入：10

輸出結果：0,2,4,6,8,10

題目分析

我們首先需要一個循環(huán)，找到 10 以內的所有數字，然后判斷是否是偶數。題目要求用生成器的方式，我們只需要寫一個生成器函數，在找到符合條件的數字后，使用 yield 輸出。最后直接循環(huán)生成器，找到需要的值放入列表，最后用逗號分隔輸出即可。

參考

def EvenGenerator(n):
   i = 0
   while i <= n:
       if i%2 == 0:
           yield i
       i += 1


n = int(input())
values = []
for i in EvenGenerator(n):
   values.append(str(i))

print(",".join(values))

問題 2

輸入一個數字 n，寫一個程序判斷 0 到 n 之間的可以同時被 5 和 7 整除的數字，并用逗號分隔來輸出。

比如輸入：100

輸出結果：

0,35,70

題目分析

此題目和上一個類似，你會發(fā)現這一次只是核心判斷條件有變化，輸入和輸出不一樣，但其實套路沒變，還是用一個生成器函數，對應條件的地方用 yield 把數據返回即可。

參考:

def NumGenerator(n):
   for i in range(n+1):
       if i%5 == 0 and i%7 == 0:
           yield i

n = int(input())
values = []
for i in NumGenerator(n):
   values.append(str(i))

print(",".join(values))

上述兩個練習都使用了 yield 把一個普通函數，變成了 generator 函數。這樣 Python 解釋器在處理這些代碼時，會在調用這個生成器的循環(huán)中，遇到 yield 就返回一次值，并保存當前變量的狀態(tài)，當下一次循環(huán)到來時，自動從上一次的狀態(tài)繼續(xù)往后計算。也就是我們需要的結果了。