Python中的迭代器

 

这里总结下Python中迭代器的一些用法，方便自己记忆，同时可供大家参考学习。

迭代器(Iterator)是访问集合内元素的一种方式。迭代器对象从集合的第一个元素开始访问，直到所有的元素都被访问一遍后结束。

迭代器不需要事先准备好整个迭代过程中所有的元素。迭代器仅仅在迭代至某个元素时才计算该元素，而在这之前或之后，元素可以不存在或者被销毁。这个特点使得它特别适合用于遍历一些巨大的或是无限的集合，如斐波那契数列。此外，迭代器还可以提供了一个统一的访问集合的接口。只要能够实现__iter__()方法的对象，都可以使用迭代器进行访问。

1.使用内建的函数iter(iterable)，可以获取迭代器对象：

>>> set = range(2)

>>> ite = iter(set)

>>> print(ite)         # 0 1

此外，使用迭代器的next()方法可以访问下一个元素： it.next()

在不清楚有多少元素可以被迭代器访问时，我们通过条件语句来做判断：

 ite = iter(set)

 try:

 while True:

    val = ite.next()

    print val

except 

    StopIteration:

pass 

2.使用迭代器的循环可以避开索引，但有时候我们还是需要索引来进行一些操作的。这时候内建函数enumerate就派上用场咯，它能在iter函数的结果前加上索引，以元组返回:

for idx, ele in enumerate(set):

    print idx, ele

 

3.生成器表达式(Generator expression)和列表解析(List Comprehension)

(x+1 for x in lst)  #生成器表达式，返回迭代器。外部的括号可在用于参数时省略。 

[x+1 for x in lst]  #列表解析，返回list