python迭代器如何实现

迭代是Python最强大的功能之一，是访问集合元素的一种方式。迭代器是一个可以记住遍历的位置的对象。迭代器对象从集合的第一个元素开始访问，直到所有的元素被访问完结束。迭代器只能往前不会后退。

迭代器有两个基本的方法：iter() 和 next()。

字符串，列表或元组对象都可用于创建迭代器：

实例(Python 3.0+)

>>>list=[1,2,3,4]
>>>it=iter(list)#创建迭代器对象
>>>print(next(it))#输出迭代器的下一个元素
1
>>>print(next(it))
2
>>>

迭代器对象可以使用常规for语句进行遍历：

实例(Python 3.0+)

#!/usr/bin/python3

list=[1,2,3,4]
it=iter(list)#创建迭代器对象
forxinit:
print(x,end="")

执行以上程序，输出结果如下：

1234

相关推荐：《Python基础教程》

也可以使用 next() 函数：

实例(Python 3.0+)

#!/usr/bin/python3
importsys#引入sys模块

list=[1,2,3,4]
it=iter(list)#创建迭代器对象

whileTrue:
try:
print(next(it))
exceptStopIteration:
sys.exit()

执行以上程序，输出结果如下：

1
2
3
4

创建一个迭代器

把一个类作为一个迭代器使用需要在类中实现两个方法 __iter__() 与 __next__() 。

如果你已经了解的面向对象编程，就知道类都有一个构造函数，Python 的构造函数为 __init__(), 它会在对象初始化的时候执行。

__iter__() 方法返回一个特殊的迭代器对象， 这个迭代器对象实现了 __next__() 方法并通过 StopIteration 异常标识迭代的完成。

__next__() 方法（Python 2 里是 next()）会返回下一个迭代器对象。

创建一个返回数字的迭代器，初始值为 1，逐步递增 1：

实例(Python 3.0+)

classMyNumbers:
def__iter__(self):
self.a=1
returnself

def__next__(self):
x=self.a
self.a+=1
returnx

myclass=MyNumbers()
myiter=iter(myclass)

print(next(myiter))
print(next(myiter))
print(next(myiter))
print(next(myiter))
print(next(myiter))

执行输出结果为：

1
2
3
4
5

StopIteration

StopIteration 异常用于标识迭代的完成，防止出现循环的情况，在 __next__() 方法中我们可以设置在完成指定循环次数后触发 StopIteration 异常来结束迭代。

在 20 次迭代后停止执行：

实例(Python 3.0+)

classMyNumbers:
def__iter__(self):
self.a=1
returnself

def__next__(self):
ifself.a<=20:
x=self.a
self.a+=1
returnx
else:
raiseStopIteration

myclass=MyNumbers()
myiter=iter(myclass)

forxinmyiter:
print(x)

执行输出结果为：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20