迭代器的坑

233倔强不秃

看小甲鱼视频时：
眼：“嗯，我看懂了。”，耳朵：“嗯，我听懂了。”，脑子：“嗯，我理解了。”
自己写的时候：
手：“？？？”

一、问题发现来源于第48讲课后题，先回顾一下：

1. 写一个迭代器，要求输出至今为止的所有闰年。如：

1. >>> leapYears = LeapYear()
   
2. >>> for i in leapYears:
   
3.         if i >=2000:
   
4.                 print(i)
   
5.         else:
   
6.                 break
   
7. 
   
8. 2012
   
9. 2008
   
10. 2004
    
11. 2000

复制代码

下面是我第一次写的时候，报错的答案：

1. '''写一个迭代器，要求输出至今为止的所有闰年'''
   
2. 
   
3. class LeapYear:
   
4.    
   
5.     def __init__(self):
   
6.         
   
7.         self.year = 2020
   
8.    
   
9.     def __next__(self):
   
10.         if ((self.year%4 == 0 and self.year%100 != 0) or (self.year%400 == 0)):
    
11.             temp = self.year
    
12.             self.year -= 1
    
13.             return temp
    
14.         else:
    
15.             self.year -1
    
16.         
    
17. 
    
18.     def __iter__(self):
    
19.         return self
    
20. 
    
21.    
    
22.    
    
23.         
    
24. 
    
25. leapYears = LeapYear()
    
26. for i in leapYears:
    
27.     if i >=2000:
    
28.             print(i)
    
29.     else:
    
30.             break
    

复制代码

我的思路是：在__next__()方法里做文章，判断如果是闰年就返回，如果不是就-1
但是却报错：

1. 2020
   
2. Traceback (most recent call last):
   
3.   File "F:/Python demo/课后作业/48.1.py", line 27, in <module>
   
4.     if i >=2000:
   
5. TypeError: '>=' not supported between instances of 'NoneType' and 'int'

复制代码

报错原因：'>='无法在'NoneType'和'int'中进行比较
所以以上代码的错误就在于：如果不是闰年，__next__()方法就返回None，这样肯定是错的。
修改需要只返回闰年，不是闰年就跳过，所以有了下面的正确答案。

1. import datetime as dt
   
2. 
   
3. class LeapYear:
   
4.     def __init__(self):
   
5.         self.year = dt.date.today().year
   
6. 
   
7.     def __iter__(self):
   
8.         return self
   
9. 
   
10.     def isLeapYear(self, year):
    
11.         if ((year%4 == 0 and year%100 != 0) or (year % 400 == 0)):
    
12.             return True
    
13.         else:
    
14.             return False
    
15. 
    
16.     def __next__(self):
    
17.         while not self.isLeapYear(self.year):
    
18.             self.year -= 1
    
19. 
    
20.         temp = self.year
    
21.         self.year -= 1
    
22.         return temp
    
23. 
    
24.    
    
25. leapYears = LeapYear()
    
26. for i in leapYears:
    
27.     if i >=2000:
    
28.             print(i)
    
29.     else:
    
30.             break
    

复制代码

二、一个小技巧
重写__next__()函数的时候，可以先定义一个变量代表索引值，看需要而定，有时候需要一个始端，一个尾端。
在return之前，一定要先将返回值存入一个中间变量，再返回中间变量，这样可以对返回值进行操作，以便于下一次执行。

1. class MyRev:
   
2. 
   
3. ##    方法一
   
4. ##    def __init__(self, arg = None):
   
5. ##        self.arg = arg
   
6. ##        self.index = 0
   
7. ##        self.len = len(arg)
   
8. ##        
   
9. ##    def __iter__(self):
   
10. ##        return reversed(self.arg)
    
11. ##
    
12. ##    def __next__(self):
    
13. ##        if self.index < self.len:
    
14. ##            val = self.age[self.index]
    
15. ##            self.index += 1
    
16. ##            return val
    
17. ##        else:
    
18. ##            raise StopIteration()
    
19. 
    
20. ##     方法二
    
21.     def __init__(self, arg = None):
    
22.         self.arg = arg
    
23.         self.index = len(arg) - 1
    
24. 
    
25.     def __iter__(self):
    
26.         return self
    
27. 
    
28.     def __next__(self):
    
29.         if self.index >= 0:
    
30.             val = self.arg[self.index]
    
31.             self.index -= 1
    
32.             return val
    
33.         else:
    
34.             raise StopIteration()
    

复制代码

蒋博文

对于你的第一段，我有深刻体会

233倔强不秃

蒋博文 发表于 2020-3-15 17:51
对于你的第一段，我有深刻体会

hhh，都是掉进过坑的

wuqramy

看小甲鱼视频时：
眼：“嗯，我看懂了。”，耳朵：“嗯，我听懂了。”，脑子：“嗯，我理解了。”
自己写的时候：
手：“？？？”

哈哈哈哈哈