timmothy 发表于 2020-4-1 16:24:45

python 48讲迭代器求教

课后作业第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
当然,你加进类里也是没有问题的

timmothy 发表于 2020-4-1 16:36:59

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



ok感谢
我以为i会一直指代的self.year
现在才知道i指代的是__next__的返回值
页: [1]
查看完整版本: python 48讲迭代器求教