码迷,mamicode.com
首页 > 其他好文 > 详细

【学习】使用生成函数实现可迭代对象

时间:2017-05-30 16:09:12      阅读:181      评论:0      收藏:0      [点我收藏+]

标签:numbers   生成器   enum   color   init   self   nbsp   tar   生成   

实现一个可迭代对象的类,它能迭代出给定范围内所有素数

实现方法:将该类的__iter__方法实现生成器函数,每次yield返回一个素数

 

 1 class PrimeNumbers:
 2     def __init__(self,start,end):
 3         self.start = start
 4         self.end = end
 5         
 6     def isPrimeNum(self,k):
 7         if k < 2:
 8             return False
 9         for i in xrange(2,k):
10             if k % i == 0:
11                 return False
12         
13         return True
14         
15     def __iter__(self):
16         for k in xrange(self.start,self.end + 1):
17             if self.isPrimeNum(k):
18                 yield k
19                 
20 if __name__ == "__main__":
21     for x in PrimeNumbers(1,100):
22         print x

 

【学习】使用生成函数实现可迭代对象

标签:numbers   生成器   enum   color   init   self   nbsp   tar   生成   

原文地址:http://www.cnblogs.com/banshaohuan/p/6919936.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!