Python 创建迭代器
创建迭代器
要创建一个对象/类作为迭代器,你必须在你的对象中实现 __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))
自己动手试一试 »