详解Python中的迭代器
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: break
3. 手动构建迭代器
把一个类作为一个迭代器使用需要在类中实现两个方法 __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: break