Python生成器

  生成器(Generator)是Python中的一类特殊对象,如果一个函数中出现一次或多次yield表达式,那么这个函数就是一个生成器。

  生成器的特点是:

  • 调用生成器时,函数体并没有执行
  • 调用生成器时,返回的是一个特殊的迭代器对象,这个迭代器对象封装了生成器的函数体、本地变量和当前执行点;
  • 当生成的迭代器的next()方法被调用时,被封装的函数体执行到下一个yield的位置,
  • 执行 yield 语句时,返回 yield 后面的表达式作为next()方法的返回值,然后把执行状态保存起来。下一次再执行 next() 方法时,函数体继续执行到下一个yield语句
  • 函数体执行结束或遇到 return 语句,则生成器抛出异常StopIteration,然后迭代过程结束
  • 生成器中的 return 语句不能带有表达式

生成器对比返回列表的函数:

  • 每次执行才返回需要的结果,比一次将所有结果放在列表中返回更节省空间
  • 如果需要同时传入所有的元素,则需要使用列表

例1. 一个生成器的例子

def updown(n):
    for x in xrange(1, n): yield x
    for x in xrange(n, 0, -1): yield x
for i in updown(5):
    print i ,
1 2 3 4 5 4 3 2 1

  

生成器表达式

  生成器表达式(generator expression)和列表推导式(list comprehension)具有非常类似的结构,列表推导式是使用方括号[]表示生成一个列表,而生成器表达式则是使用圆括号()返回一个一次产生一个值的生成器:

 

例:生成器表达式与列表推导式

>>> l = range(1,10)
>>> l
[1, 2, 3, 4, 5, 6, 7, 8, 9]
>>> [x*x for x in l] #列表推导式
[1, 4, 9, 16, 25, 36, 49, 64, 81]
>>> (x*x for x in l) # 生成器表达式
<generator object <genexpr> at 0x000000000224E288>

  显然,生成器表达式兼具列表推导式的简洁的优点,还能节省内存。

posted @ 2015-08-08 20:05  王智愚  阅读(365)  评论(0编辑  收藏  举报