Python之迭代器

定义

可以被next()函数调用并不断返回下一个值的对象称为迭代器(Iterator对象)。

迭代器是访问集合元素的一种方式。迭代器对象从集合的第一个元素开始访问，直到所有的元素被访问完结束。迭代器只能往前不会后退。

特点

迭代器不要求你事先准备好整个迭代过程中所有的元素。仅仅是在迭代至某个元素时才计算该元素，而在这之前或之后，元素可以不存在或者被销毁。这个特点使得它特别适合用于遍历一些巨大的或是无限的集合。

迭代器不但可以作用于for循环，还可以被next()函数不断调用并返回下一个值，直到最后抛出一个StopIteration错误表示无法继续返回下一个值。

应用

迭代器的使用需导入相关模块。

from collections import Iterable
from collections import Iterator

可以用isinstance()判断一个对象是否是Iterator对象。

print(isinstance((), Iterator))
print(isinstance([], Iterator))
print(isinstance({}, Iterator))
print(isinstance("", Iterator))
print(isinstance((x for x in range(10)), Iterator))

可迭代对象

迭代器提供了一个统一的访问集合的接口。只要是实现了__iter__()或__getitem__()方法的对象，就可以使用迭代器进行访问。

可以直接作用于for循环的对象统称为可迭代对象(Iterable)。一般分为两种：集合数据类型(包括list, tuple,dict.set或string)和generator(包括生成器和带yield的generator function)。

序列：字符串、列表、元组

非序列：字典、文件

自定义类：用户自定义的类实现了__iter__()或__getitem__()方法的对象

可以用isinstance()判断一个对象是否是Iterable对象。

print(isinstance((), Iterable))
print(isinstance([], Iterable))
print(isinstance({}, Iterable))
print(isinstance("", Iterable))
print(isinstance((x for x in range(10)), Iterable))

创建迭代器

i = (x for x in range(5))  # 直接定义一个迭代器i
print(next(i))  # i可以被next一直调用下去

l = [1,2,3,4,5]
i = iter(l)  # list直接转成迭代器i
i = iter((1,2,3,4,5))  # tuple直接转成迭代器i

案例

使用迭代器实现输入字符串，并且回车键换行继续输入字符串，以end结束输入。

endstr =  "end"
str = ""

for line in iter(input, endstr):
    str += line + "\n"

print(str)