使用 __iter__ 和 __next__ 创建我自己的类

Question

我正在尝试为迭代器创建自己的类并找到了这样一个示例：

class OddNum:
  """Class to implement iterator protocol"""

  def __init__(self, num = 0):
    self.num = num

  def __iter__(self):
    self.x = 1
    return self

  def __next__(self):
    if self.x <= self.num:
      odd_num = self.x
      self.x += 2
      return odd_num
    else:
      raise StopIteration

for num in OddNum(10):
  print(num)

输出为：1 3 5 7 9

现在，如果我删除行 odd_num = self.x 并将 return odd_num 更改为 return self.x，我会得到以下输出：3 5 7 9 11

这2个代码有什么区别，为什么要定义一个变量为self.x？

Answer 1

在这种特定情况下，发生的情况是，在为下一次迭代准备好 self.x 的值递增之前，您希望从 __next__ 方法中间的某个位置返回 self.x 的值。因此变量odd_num 用于保存该值。相反，如果您在 self.x += 2 语句之后返回 self.x（即添加 2），那么您将得到不同的答案，如您所见。

您可以考虑的另一种可能性是，如果您想要使用代码中间某处某个变量的值，而不是编写自己的类，而是使用生成器函数来实现迭代器。这是因为您可以将yield 声明放在您喜欢的任何位置。所以在这个例子中，它看起来像这样：

def odd_num(num=0):
    x = 1
    while x <= num:
        yield x
        x += 2
  
    
for num in odd_num(10):
    print(num)

这给出了：

1
3
5
7
9

（注意while 代替了您在__next__ 方法中使用的if。）

这里，yield 语句在 x += 2 语句之前，这不是使用方法（函数）中的return 可以完成的。所以不需要将值保存在另一个变量中以供以后使用。