python中的生成器

yupeng發表於2013-10-11
Python

什麼是生成器？

　　生成器是一個包含了特殊關鍵字yield的函式。當被呼叫的時候，生成器函式返回一個生成器。可以使用send，throw，close方法讓生成器和外界互動。

生成器也是迭代器，但是它不僅僅是迭代器，擁有next方法並且行為和迭代器完全相同。所以生成器也可以用於python的迴圈中，

如何使用？

首先看一個例子：

 1 #!/usr/bin/python
 2 # -*- coding: utf-8 -*-
 3 
 4 def flatten(nested):
 5     for sublist in nested:
 6         for element in sublist:
 7             yield element
 8 
 9 nested = [[1,2],[3,4],[5,6]]
10 
11 for num in flatten(nested):
12     print num,

結果為1，2，3，4，5，6

遞迴生成器：

 1 #!/usr/bin/python
 2 # -*- coding: utf-8 -*-
 3 
 4 def flatten(nested):
 5     try:
 6         for sublist in nested:
 7             for element in flatten(sublist):
 8                 yield  element
 9     except TypeError:
10         yield nested
11 
12 for num in flatten([[1,2,3],2,4,[5,[6],7]]):
13     print num

結果為：1 2 3 2 4 5 6 7

讓我們一起來看看生成器的本質

首先看下：

 1 #!/usr/bin/python
 2 # -*- coding: utf-8 -*-
 3 
 4 def simple_generator():
 5     yield 1
 6 
 7 print simple_generator
 8 
 9 def repeater(value):
10     while True:
11         new  = (yield value)
12         if new is not None: value = new
13 
14 
15 r = repeater(42)
16 print r.next()
17 
18 print r.send('hello,world!')

結果為：

<function simple_generator at 0x10c76f6e0>
42
hello,world!

可以看出：

1）生成器就是一函式

2）生成器具有next方法

3）生成器可以使用send 方法和外界互動。

---end---

 

 

　　　　

相關文章