Python的yield关键字

忽然得知Python有个叫yield的关键字，好奇之下去查了查，花了点时间基本弄明白了，故写在此备忘，顺便充充字数。

yield关键字用来定义生成器（Generator），其具体功能是可以当return使用，从函数里返回一个值，不同之处是用yield返回之后，可以让函数从上回yield返回的地点继续执行。也就是说，yield返回函数，交给调用者一个返回值，然后再“瞬移”回去，让函数继续运行， 直到吓一跳yield语句再返回一个新的值。

使用yield返回后，调用者实际得到的是一个迭代器对象，迭代器的值就是返回值，而调用该迭代器的next()方法会导致该函数恢复yield语句的执行环境继续往下跑，直到遇到下一个yield为止，如果遇不到yield，就会抛出异常表示迭代结束。

看一个例子：

>>> def test_yield():
...     yield 1
...     yield 2
...     yield (1,2)
...
>>> a = test_yield()
>>> a.next()
1
>>> a.next()
2
>>> a.next()
(1, 2)
>>> a.next()
Traceback (most recent call last):
  File "<stdin>", line 1, in ?
StopIteration

 

每次调用a.next()，都会得到下一个yield返回的值，最后一次调用会抛出StopIteration结束异常。

 

暂时没想到这个东西有什么用，也许就是可以省一些代码。

 

一个有趣的例子，用来生成一个序列的全排列：

def


 perm(


items, n=None


)


:



    if


 n is


 None


:



        n = len


(


items)






    for


 i in


 range


(


len


(


items)


)


:



        v = items[


i:i+1]






        if


 n == 1:



            yield


 v



        else


:



            rest = items[


:i]


 + items[


i+1:]






            for


 p in


 perm(


rest, n-1)


:



                yield


 v + p










的确是方便一些，不过我还是觉得都放到一个数组里更直观...