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))
亲自试一试 »