Python 实现 itertools.count

永恒的蓝色梦想

1. from numbers import Number
   
2. 
   
3. def count(start, step=1):
   
4.     if isinstance(start,Number) and isinstance(step,Number):
   
5.         while True:
   
6.             yield start
   
7.             start+=step
   
8. 
   
9.     else:
   
10.         raise TypeError("a number is required")

复制代码

如果代码有问题，欢迎在评论区指出

zltzlt

不应该是一个类吗

永恒的蓝色梦想

zltzlt 发表于 2020-4-2 18:59
不应该是一个类吗

其实差别也不大，全凭心情

1. from numbers import Number
   
2. 
   
3. class count:
   
4.     def __init__(self,start,step=1):
   
5.         if isinstance(start,Number) and isinstance(step,Number):
   
6.             self.__start=start
   
7.             self.__step=step
   
8. 
   
9.         else:
   
10.             raise TypeError("a number is required")
    
11.    
    
12.     def __iter__(self):
    
13.         return self
    
14. 
    
15.     def __next__(self):
    
16.         temp=self.__start
    
17.         self.__start+=self.__step
    
18.         return temp

复制代码

小人物来了

看不懂，来学习