a sample example to understand and use yield

yield 用法说明



yield 简单来说就是一个生成器,生成器是这样一个函数,它记住上一次返回时在函数体中的位置。对生成器函数的第二次(n次)调用跳转至该函数中间,而上次调用的所有局部变量都保持不变。



  • 生成器是一个函数,函数的所有参数都会保留

  • 第二次调用此函数时使用的参数是前一次保留下的

  • 生成器还记住了它在流控制构造

yield 运行机制



当生成器要一个数时,生成器会执行,直至出现yield语句,生成器返回yield参数,之后生成器停止继续执行。
当需要下一个数时,生成器从上次的状态开始执行,直至出现yield语句,返回参数,停止。如此反复。

例子:



#生成全排列
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

#生成组合
def comb(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+1:]
for c in comb(rest,n-1):
yield v+c

a=perm(‘abc’)
for b in a:
print b
break
print ‘=’*20
for b in a:
print b



结果如下:


102 pvopf006 ~/test> ./generator.py

abc

acb
bac
bca
cab
cba



可以看到,在第一个循环break后,生成器没有继续执行,而第二个循环接着第一个循环执行