关于iter()方法

hangfrieddays

想问关于iter方法的第二种使用，我以下的代码是否get到了精髓，或者说是多此一举呢

1. class Iter:
   
2.     def __init__(self):
   
3.         self.count = 0
   
4. 
   
5.     def __iter__(self):
   
6.         return self
   
7. 
   
8.     def __next__(self):
   
9.         self.count += 1
   
10.         if self.count > 3:
    
11.             raise StopIteration
    
12.         return self.count
    
13. 
    
14. 
    
15. a = Iter()
    
16. b = iter(a.__next__, 10)
    
17. print(next(b))

复制代码

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.    
    

复制代码