python学习笔记之八：迭代器和生成器

风起漂泊

　　一. 迭代器
　　在前面的笔记中，已经提到过迭代器（和可迭代），这里会对此进行深入讨论。只讨论一个特殊方法---__iter__，这个方法是迭代器规则的基础。
　　1.1 迭代器规则
　　迭代的意思是重复做一些事情很多次，到现在为止只是在for循环中对序列和字典进行迭代，但实际上也能对其他的对象进行迭代：实现__iter__方法的对象。
　　__iter__方法返回一个迭代器，所谓迭代器就是具有next方法（这个方法在调用时不需要任何参数）的对象。在调用next方法时，迭代器会返回它的下一个值。如果next方法被调用，但迭代器没有值可以返回，就会引发一个StopIteration异常。
　　为什么不用列表？列表是一次性获取所有值的，如果有很多值，列表会占用太多的内存，但是迭代器可以在使用时计算一个值时获取一个值，使用迭代器更通用，简单，优雅。
　　下面来看一个不使用列表的例子：



__metaclass__ = type
class Fibs:
def __init__(self):
self.a = 0
self.b = 0
def next(self):
self.a,self.b = self.b,self.a + self.b
return self.a
def __iter__(self):
return self
　　注意：迭代器实现了__iter__方法，这个方法实际上返回迭代器本身。很多情况下，__iter__会放到其他的会在for循环中使用的对象中。这样一来，程序就能返回所需的迭代器。此外，推荐使用迭代器实现它自己的_iter__方法，然后就能直接在for循环中使用迭代其本身了。



>>> f = Fibs()
>>> for n in f:
if n > 10:
print n
break
13
　　
　　1.2 从迭代器得到序列
　　除了在迭代器和可迭代对象上进行迭代外，还能把它们转换为序列。在大部分能使用序列的情况下，能使用迭代器替换。关于这个的一个很有用的例子是使用list构造方法显式地将迭代器转化为列表。



__metaclass__ = type
class TestIterator:
value = 0
def next(self):
self.value += 1
if self.value > 10:raise StopIteration
return self.value
def __iter__(self):
return self


>>> ti = TestIterator()
>>> list(ti)
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
　　
　　 二. 生成器
　　生成器是python新引入的概念，由于历史原因，它也叫简单生成器。它和迭代器可能是近几年来引入的最强大的两个特性。生成器可以帮助读者写出非常优雅的代码。
　　生成器是一种用普通的函数语法定义的迭代器。让我们先看看怎么创建和使用生成器，然后再了解一下她的内部机制。
　　2.1 创建生成器
　　首先创建一个展开嵌套列表的函数，参数是一个列表的列表，类似于nested = [[1,2],[3,4],[5]]，函数按照顺序打印出列表中的数字：



__metaclass__ = type
def flatten(nested):
for sublist in nested:
for element in sublist:
yield element
　　这个函数首先迭代提供的嵌套列表中的所有子列表，然后按顺序迭代列表中的元素。yield语句是新知识，任何包含yield语句的函数都称为生成器。它不像return那样返回值，而是每次产生多个值。每次产生一个值，函数就会被冻结：即函数停在那点等待被激活，函数被激活后就从停止的那点开始执行。
　　接下来可以通过在生成器上迭代来使用所有的值：



>>> nested = [[1,2],[3,4],[5]]
>>> for num in flatten(nested):
print num
1
2
3
4
5
　　
　　 2.2 递归生成器
　　上面的例子中只能处理两层嵌套，如果不知道有多少层嵌套呢？我们就要求助于递归了：



__metaclass__ = type
def flatten(nested):
try:
for sublist in nested:
for element in sublist:
yield element
except TypeError:
yield nested
　　当flatten被调用时，有两种可能性：基本情况和需要递归的情况，在特殊情况中，展开的是一个列表，程序必须遍历所有的字列表，并对他们调用flatten。然后用另一个for循环来产生被展开的子列表中的所有元素。



>>> nested = [[1,2],[3,4],[5,[6,[7]]]]
>>> for num in flatten(nested):
    print num
1
2
3
4
5
[6, [7]]
　　（上面的结果跟书《python基础教程（第二版）》中例子的结果不一样，我也搞不懂）