Python 创建迭代器
创建迭代器
要创建一个对象/类作为迭代器,您必须在对象中实现 iter()
和 next()
方法。
正如您在 Python 类/对象章节中所学到的,所有类都有一个名为 init()
的函数,它允许您在创建对象时进行一些初始化操作。
iter()
方法的作用类似,您可以执行一些操作(初始化等),但必须始终返回迭代器对象本身。
next()
方法还允许您执行操作,并且必须返回序列中的下一个项目。
实例
创建一个迭代器,从 1 开始返回数字,并且每个序列将递增 1(返回 1,2,3,4,5 等):
class MyNumbers: def __iter__(self): self.a = 1 return self def __next__(self): x = self.a self.a += 1 return x myclass = MyNumbers() myiter = iter(myclass) print(next(myiter)) print(next(myiter)) print(next(myiter)) print(next(myiter)) print(next(myiter))