生成器generator:
生成器不需要一次性将对象的所有元素都存入内存之后,才开始进行操作。生成器仅在迭代至某个元素时才会将该元素放入内存,而在这之前或之后,元素可以不存在或者被销毁。
python有两种不同的方式提供生成器:
一、生成器表达式:
把一个列表生成式的 [ ] 改成( ),就创建了一个generator生成器。
类似于列表生成式(一次性构建好列表放入内存用于迭代)。但是,生成器表达式并不是一次性在内存中构建一个完整的列表,它返回的是一个生成器对象,并且它是一边循环一边推算列表后续的元素。生成器对象每次按需产生结果。
g=(i+1 for i in range(0,10,2))
print g
要获得生成器的每一个元素:
1、 可以通过next(生成器对象)函数获得生成器的下一个返回值。每次调用next(),就计算出生成器对象的下一个元素的值,直到计算到最后一个元素,没有更多的元素时,抛出StopIteration的错误。
g=(i+1 for i in range(0,10,2))
print next(g)
print next(g)
print next(g)
print next(g)
print next(g)
print next(g)
2、 也可以通过for循环迭代
g=(i+1 for i in range(0,10,2))
for i in g:
print i
二、生成器函数:
定义与常规函数一致,但,返回结果通过yield语句而不是return语句。yield语句一次返回一个结果,在每个结果中间,挂起函数的状态,便于下次从它中断的地方继续执行。可以通过next(生成器对象)函数获得生成器函数的下一个返回值。
在函数的执行过程中,执行完yield语句后,会退出函数,然后把需要的值返回给调用生成器函数的地方,下一次调用生成器函数的时候又从上次中断的地方开始执行,而生成器内的所有变量参数都会被保存下来供下一次使用。
生成器函数的执行流程:在每次调用next()的时候执行,遇到yield语句就中断并返回,再次执行时从上次执行到的yield语句处继续向下执行。调用生成器函数前,首先要生成一个生成器对象,再通过next()函数执行生成器函数获得返回值。
1、 可以通过next(生成器对象)函数获得生成器函数的下一个返回值。当超过了返回值数量会抛出StopIteration的错误
# -*- coding: cp936 -*-
def printnum():
print"第一次执行函数的返回值:"
yield 3
print"第二次执行函数的返回值:"
yield 2
print"第三次执行函数的返回值:"
yield 1
print"第四次执行函数的返回值:"
yield "The END~"
g=printnum()
print next(g)
print next(g)
print next(g)
print next(g)
print next(g)
2、 也可以通过for循环迭代
# -*- coding: cp936 -*-
def printnum():
print"第一次执行函数的返回值:"
yield 3
print"第二次执行函数的返回值:"
yield 2
print"第三次执行函数的返回值:"
yield 1
print"第四次执行函数的返回值:"
yield "The END~"
for i in printnum():
print i