python使用產生器實現可迭代對象，python產生器

python使用產生器實現可迭代對象，python產生器

本文執行個體為大家分享了python使用產生器實現可迭代對象的具體代碼，供大家參考，具體內容如下

案例分析：      

        實一個可迭代對象的類，它能迭代出給定範圍內所有的素數：

              pn = Number(1, 30)

              for k in pn:

                     print(k)

       結果為：2,3,5,7，11,13，17,19,23，29

如何解決這個問題？

將該類的__iter__方法實現成產生器函數，每次yield返回一個素數

#!/usr/bin/python3  class Number(object): def __init__(self, start, end):  self.start = start  self.end = end    # 判斷一個數字是否是素數 def get_num(self, k):  if k >= 2:   for i in range(2, k):    if k % i == 0:     return False   return True   def __iter__(self):  for k in range(self.start, self.end+1):   if self.get_num(k):    # 是素數yield出去    yield k     if __name__ == '__main__': num = Number(2, 30) for i in num:  print(i)

以上就是本文的全部內容，希望對大家的學習有所協助，也希望大家多多支援幫客之家。