前面介绍了使用 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
原创文章,作者:奋斗,如若转载,请注明出处:https://blog.ytso.com/tech/pnotes/22609.html