[已解决]python 48讲迭代器求教

timmothy

课后作业第1题，要求输出至今为止的所有闰年
我对迭代器不是很懂，写了以下的程序
class LeapYear:
    def __init__(self):
        self.year = 2020
   
    def __iter__(self):
        return self
   
    def __next__(self):
        self.year -= 1
        if ((self.year%4 == 0 and self.year%100 != 0) or (self.year%400 == 0)):
            return self.year

然后在用下述代码测试的时候报错：'>=' not supported between instances of 'NoneType' and 'int'
>>> leapYears = LeapYear()
>>> for i in leapYears:
        if i >=2000:
                print(i)
        else:
                break

为什么这个i是NoneType的量，难道不是指代的self.year嘛？

最佳答案

月排行榜 / 总排行榜

qiuyouzhi

2020-4-1 16:31:36

因为你的代码如果不是闰年就会返回None，
所以这里加一个判断：

>>> leapYears = LeapYear()
>>> for i in leapYears:
        if i == None:
                continue
        elif i >= 2000:
                print(i)
        else:
                break

        
2016
2012
2008
2004
2000

当然，你加进类里也是没有问题的

跳转到最佳答案楼层

qiuyouzhi

因为你的代码如果不是闰年就会返回None，
所以这里加一个判断：

>>> leapYears = LeapYear()
>>> for i in leapYears:
        if i == None:
                continue
        elif i >= 2000:
                print(i)
        else:
                break

        
2016
2012
2008
2004
2000

当然，你加进类里也是没有问题的

timmothy

qiuyouzhi 发表于 2020-4-1 16:31
因为你的代码如果不是闰年就会返回None，
所以这里加一个判断：

ok感谢
我以为i会一直指代的self.year
现在才知道i指代的是__next__的返回值