Python __iter__和__reversed__:实现迭代器

广告位

前面介绍了使用 for 循环遍历列表、元姐和字典等,这些对象都是司法代的,因此它们都属于迭代器。 如果开发者需…

前面介绍了使用 for 循环遍历列表、元姐和字典等,这些对象都是司法代的,因此它们都属于迭代器。

如果开发者需要实现法代器,只要实现如下两个方法即可:

  • __iter__(self):该方法返回一个法代器(iterator),迭代器必须包含一个__next__()方法,该方法返回迭代器的下一个元素。
  • __reversed__(self):该方法主要为内建的reversed()反转函数提供支持,当程序调用 reversed() 函数对指定迭代器执行反转时,实际上是由该方法实现的。

从上面介绍不难看出,如果程序不需要让迭代器反转迭代,其实只需要实现第一个方法即可。下面程序将会定义一个代表斐波那契数列(数列的元素等于前两个元素之和:f(n+2) = f(n+1) +f(n))的迭代器。

  # 定义一个代表斐波那契数列的迭代器  class Fibs:      def __init__(self, len):          self.first = 0          self.sec = 1          self.__len = len      # 定义迭代器所需的__next__方法      def __next__(self):          # 如果__len__属性为0,结束迭代          if self.__len == 0:              raise StopIteration          # 完成数列计算:          self.first, self.sec = self.sec, self.first + self.sec          # 数列长度减1          self.__len -= 1          return self.first      # 定义__iter__方法,该方法返回迭代器      def __iter__(self):          return self  # 创建Fibs对象  fibs = Fibs(10)  # 获取迭代器的下一个元素  print(next(fibs))  # 使用for循环遍历迭代器  for el in fibs:      print(el, end=' ')

上面程序定义了一个 Fibs 类,该类实现了 __iter__() 方法,该方法返回 self,因此它要求该类必须提供 __next__() 方法,该方法会返回数列的下二个值。程序使用 len 属性控制数列的剩余长度,当 __len 为 0 时,程序停止遍历。
 
上面程序创建了一个长度为 10 的数列,程序开始使用内置的 next() 函数来获取迭代器的下一个元素,该 next() 函数其实就是通过迭代器的 __next()__ 方法来实现的。

程序接下来使用 for 循环来遍历应数列。运行该程序,将看到如下输出结果:

1 1 2 3 5 8 13 21 34 55

此外,程序可使用内置的 iter() 函数将列表、元组等转换成迭代器,例如如下代码:

  # 将列表转换为迭代器  my_iter = iter([2, 'fkit', 4])  # 依次获取迭代器的下一个元素  print(my_iter.__next__()) # 2  print(my_iter.__next__()) # fkit

关于作者: Python基础入门教程

为您推荐

广告位

发表评论