关于iter()方法

suchocolate

__next__里定死了3，外面用10没有意义。不如这样：

1. class Iter:
   
2.     def __init__(self, limit=1):
   
3.         self.count = 0
   
4.         self.limit = limit
   
5. 
   
6.     def __iter__(self):
   
7.         return self
   
8. 
   
9.     def __next__(self):
   
10.         self.count += 1
    
11.         if self.count > self.limit:
    
12.             raise StopIteration
    
13.         return self.count
    
14. 
    
15. 
    
16. a = Iter(3)
    
17. try:
    
18.     while True:
    
19.         print(next(a))
    
20. except Exception as e:
    
21.     print(e)
    
22. print('=' * 20)
    
23. a = Iter(10)
    
24. try:
    
25.     while True:
    
26.         print(next(a))
    
27. except Exception as e:
    
28.     print(e)
    
29.    
    

复制代码