python使用生成器實(shí)現(xiàn)可迭代對象

發(fā)布時間：2020-10-10 07:56:50 來源：腳本之家閱讀：108 作者：北門吹雪欄目：開發(fā)技術(shù)

本文實(shí)例為大家分享了python使用生成器實(shí)現(xiàn)可迭代對象的具體代碼，供大家參考，具體內(nèi)容如下

案例分析：

實(shí)一個可迭代對象的類，它能迭代出給定范圍內(nèi)所有的素數(shù)：

pn = Number(1, 30)

for k in pn:

print(k)

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

如何解決這個問題？

將該類的__iter__方法實(shí)現(xiàn)成生成器函數(shù)，每次yield返回一個素數(shù)

#!/usr/bin/python3
 
 
class Number(object):
 def __init__(self, start, end):
  self.start = start
  self.end = end
   
 # 判斷一個數(shù)字是否是素數(shù)
 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):
    # 是素數(shù)yield出去
    yield k
     
if __name__ == '__main__':
 num = Number(2, 30)
 for i in num:
  print(i)

以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持億速云。

向AI問一下細(xì)節(jié)

python使用生成器實(shí)現(xiàn)可迭代對象

猜你喜歡

最新資訊

相關(guān)推薦

相關(guān)標(biāo)簽