1. 什么是迭代器
迭代器是一个可以记住遍历位置的对象。可以遍历诸如列表，字典及字符串等序列对象
迭代器的特点：
- 迭代器对象从集合的第一个元素开始访问，直到所有的元素被访问完结束。
- 迭代器只能往前不会后退。
2. 两个基本的方法：iter() 和 next()。
序列对象可以利用iter()直接创建迭代器，并通过next()获取迭代器中的下一个元素。
# 创建迭代器对象
S = 'PYTHON'
str_iter = iter(S)
print(type(str_iter))# 利用for循环迭代
S = 'PYTHON'
str_iter = iter(S)
for word in str_iter:
print(word)# 利用next() 迭代
S = 'PYTHON'
str_iter = iter(S)
print(next(str_iter))
print(next(str_iter))
print(next(str_iter))
print(next(str_iter))
print(next(str_iter))
print(next(str_iter))# 利用 while 循环迭代
当我们使用 next() 迭代时，如果迭代次数超过了迭代器中的元素个数就会引发StopIteration 异常，由此我们可以利用while循环迭代，并不断捕捉迭代结束的异常完成for循环的迭代过程。
S = 'PYTHON'
str_iter = iter(S)
while True:
try:
print(next(str_iter))
except StopIteration:
break3. 手动构建迭代器
把一个类作为一个迭代器使用需要在类中实现两个方法 __iter__() 与 __next__() 。
__iter__() 方法返回一个特殊的迭代器对象， 这个迭代器对象实现了 __next__() 方法并通过 StopIteration 异常标识迭代的完成。
__next__() 方法（Python 2 里是 next()）会返回下一个迭代器对象。
下面这个类构造了一个迭代器接受一个可迭代的数字，每次迭代都返回上一次迭代结果的平方，当迭代结果大于9999999999999时抛出异常。利用该类创建示例对数字2进行平方和迭代。
class 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
# 直接调用
num = Square(2)
while True:
try:
print(num.__next__())
except StopIteration:
break
# 通过循环调用
for i in num:
print(i)
# 通过next()调用
num = Square(2)
my_iter = iter(num)
print(next(my_iter))
print(next(my_iter))
print(next(my_iter))
print(next(my_iter))
print(next(my_iter))
# 通过while循环调用
num = Square(2)
my_iter = iter(num)
while True:
try:
print(next(my_iter))
except StopIteration:
break12
评论
共0条评论
- 这篇文章还没有收到评论，赶紧来抢沙发吧~
