python生成器的使用方法

生成器也是迭代器，但是它不仅仅是迭代器，拥有next方法并且行为和迭代器完全相同。所以生成器也可以用于python的循环中，

生成器如何使用？

首先看一个例子：

代码如下:

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

递归生成器：

代码如下:

def flatten(nested):
try:
for sublist in nested:
for element in flatten(sublist):
yield element
except TypeError:
yield nested

for num in flatten([[1,2,3],2,4,[5,[6],7]]):
print num

结果为：1 2 3 2 4 5 6 7

让我们一起来看看生成器的本质

首先看下：

代码如下:

def simple_generator():
yield 1

print simple_generator

def repeater(value):
while True:
new = (yield value)
if new is not None: value = new

r = repeater(42)
print r.next()

print r.send('hello,world!')

结果为：
代码如下: