**
迭代器
**
迭代器是访问集合元素的一种方式。迭代器对象从集合中第一个元素开始访问,直到所有的元素被访问完结束。迭代器只能往前不会后退。
迭代器的两个方法:
iter()方法:返回迭代器对象本身
next()方法:返回容器的下一个元素,在结尾时引发StopIteration异常。
对于可迭代对象,可以使用内建函数iter()来获取它的迭代器对象:
list=[1,2]
iter=iter(list)
print iter.next()
print iter.next()
print iter.next()
打印出:
1
2
Traceback (most recent call last):
File “C:/Users/Administrator/Desktop/iter3.py”, line 5, in
print iter.next()
StopIteration
自定义迭代器
下面定义了一个MyRange类,这个类型中实现了iter()方法,通过这个方法返回对象本身作为迭代器对象;同时,实现了next()方法用来获取容器中的下一个元素,当没有可访问元素后,就抛出StopIteration异常。
class MyRange(object):
def __init__(self,n):
self.idx=0
self.n=n
def __iter__(self):
return self
def next(self):
if self.idx<self.n:
val=self.idx
self.idx+=1
return val
else:
raise StopIteration()
if __name__=="__main__":
myRange=MyRange(3)
for i in myRange:
print i
打印结果为:
0
1
2
因为myRange这个对象就是一个可迭代对象,同时它本身也是一个迭代器对象,所以就出现了问题,他没有办法支持多次迭代,例如:
class MyRange(object):
def __init__(self,n):
self.idx=0
self.n=n
def __iter__(self):
return self
def next(self):
if self.idx<self.n:
val=self.idx
self.idx+=1
return val
else:
raise StopIteration()
if __name__=="__main__":
myRange=MyRange(3)
print myRange is iter(myRange)
print [i for i in myRange]
print [i for i in myRange]
打印结果为:
True
[0, 1, 2]
[]
为了解决上面的问题,可以分别定义可迭代类型对象和迭代器类型对象;然后可迭代类型对象的iter()方法可以获得一个迭代器类型的对象。
class Zrange:
def __init__(self,n):
self.n=n
def __iter__(self):
return MyRange(self.n)
class MyRange(object):
def __init__(self,n):
self.idx=0
self.n=n
def __iter__(self):
return self
def next(self):
if self.idx<self.n:
val=self.idx
self.idx+=1
return val
else:
raise StopIteration()
if __name__=="__main__":
zrange=Zrange(3)
print zrange is iter(zrange)
print [i for i in zrange]
print [i for i in zrange]
打印结果为:
False
[0, 1, 2]
[0, 1, 2]