Python学习教程（Python学习路线）：Python 3—手动创建迭代器

千锋Python学堂 2019-06-04

展开全文

今天带大家学习Python 3如何手动创建迭代器！

Python学习教程（Python学习路线）：Python3—手动创建迭代器

概述

迭代器即可以遍历诸如列表，字典及字符串等序列对象甚至自定义对象的对象，其本质就是记录迭代对象中每个元素的位置。迭代过程从第一个元素至最后一个元素，且过程不能回滚或反方向迭代。

两个基本方法iter、next

序列对象可以利用iter()直接创建迭代器，并通过next()即可迭代迭代器。

利用for循环迭代

S = 'PYTHON'
IT = iter(S)
for it in IT:
 print(it)

示例结果：

P
Y
T
H
O
N

利用next()迭代

S = 'PYTHON'
IT = iter(S)
print(next(IT))
print(next(IT))
print(next(IT))
print(next(IT))
print(next(IT))
print(next(IT))

示例结果：

P
Y
T
H
O
N

当我们使用next()迭代时，如果迭代了次数超过了迭代器中的元素个数就会引发StopIteration，由此我们可以利用while循环迭代，并不断捕捉迭代结束的异常完成for循环的迭代过程。

S = 'PYTHON'
IT = iter(S)
while True:
 try:
 print(next(IT))
 except StopIteration:
 break
P
Y
T
H
O
N

手动构建迭代器

把一个类作为一个迭代器使用需要在类中实现两个方法 iter() 与 next() 。iter() 方法返回一个特殊的迭代器对象，这个迭代器对象实现了 next() 方法并通过 StopIteration 异常标识迭代的完成。next() 方法（Python 2 里是 next()）会返回下一个迭代器对象。下面这个类构造了一个迭代器接受一个可迭代的数字，每次迭代都返回上一次迭代结果的平方，当迭代结果大于9999999999999时抛出异常。利用该类创建示例对数字2进行平方和迭代。

class IT_SQUARE:
 def __init__(self, x):
 self.x = x
 def __next__(self):
 self.x = self.x ** 2
 if self.x > 9999999999999:
 raise StopIteration
 
 else:
 return self.x
 def __iter__(self):
 return self
 
IT1 = IT_SQUARE(2)
 
while True:
 try:
 print(IT1.__next__())
 except StopIteration:
 break