什么是生成器?
生成器是一个包含了特殊关键字yield的函数。当被调用的时候,生成器函数返回一个生成器。可以使用send,throw,close方法让生成器和外界交互。
生成器也是迭代器,但是它不仅仅是迭代器,拥有next方法并且行为和迭代器完全相同。所以生成器也可以用于python的循环中,
如何使用?
首先看一个例子:
#!/usr/bin/python
# -*- coding: utf-8 -*-
def flatten(nested):
for sublist in nested:
for element in sublist:
yield element
nested = [[1,2],[3,4],[5,6]]
for num in flatten(nested):
print num,
结果为1,2,3,4,5,6
递归生成器:
#!/usr/bin/python
# -*- coding: utf-8 -*-
def flatten2(nested):
try:
for sublist in nested:
for element in flatten2(sublist):
yield element
except:
yield nested
for num in flatten2([[1,2,3],2,4,[5,[6],7]]):
print num,
结果为:1 2 3 2