1、定义一个迭代器
要创建一个对象/类作为迭代器,必须实现__iter__()
和__next__()
方法。
正如在Python 类和对象一章中所了解的那样,所有类都有一个名为__init__()
的函数,该函数可以在创建对象时进行一些初始化。
__iter__()
方法的行为类似,可以执行操作(初始化等),但必须始终返回迭代器对象本身。
__next__()
方法还允许进行操作,并且必须返回序列中的下一项。
例如:
创建一个返回数字(从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))
相关文档: