Python生成器

生成器(generator)在Python中是为了解决内存不足而提出的一种方法,例如,我们要创建一个包含10万个元素的列表,这个列表不仅占用很大的存储空间,而且如果我们只需要访问前面几个元素,后面元素占用的空间就白白浪费了.于是Python中给出一种一边循环一边计算的机制,这就是生成器.

创建生成器的方法(genterator)

1.只要把一个列表生成式的[]改成(),就创建了一个generator

L=[x*x for x in range(5)]

g=(x*x for x in range(5))

g就是一个生成器.打印生成器的每一个元素,如果要一个一个打印出来,可以通过next()函数获取generator的下一个返回值:

next(g)

使用for循环,因为generator是一个可迭代对象:

g=(x*x for x in range(5))

for n in g:

        print(n)

2.在函数定义中包含yield关键字,这样该函数就不再是一个普通函数,而是一个generator:

def fib(max):

      n,a,b=0,0,1

       while n<max:

              yield b

               a,b=b,a+b

                n=n+1

       return  'done'