生成器(generator)的作用是一次产生一个数据项,并把数据项输出,generator函数可以用在for循环中遍历.generator函数定义:
def 函数名(参数): ... yield 表达式 |
generator函数的定义和普通函数的定义没什么区别,只要在函数体内使用yield生成数据项即可.generator函数可以被for循环遍历,并且可以通过next()方法获得yield生成的数据项
generator函数实例:
#/usr/bin/python #-*- coding:utf-8 -*- #定义generator函数 def func(n): for i in range(n): yield i for i in func(3): print i print "************************************" r=func(3) print r.next() print r.next() print r.next() print r.next() 运行结果: 0 1 2 ************************************ 0 1 2 Traceback (most recent call last): File "yieldpy.py", line 15, in ? print r.next() StopIteration |
r.next()输出yield生成的数据项,由于"n"的值等于3,因此生成3个数字后,已经没有数据生成,Python解释器抛出异常StopIteration
yield保留字与return语句的返回值和执行原理都不相同,yield生成值并不会中止程序的执行,返回值后程序继续往后执行,return返回值后,程序将中止执行