怎么自定义一个生成器:
两个方法:
1、小括号包裹表达式
2、函数中用yield返回
方法一:①小括号包裹表达式
G=(x*2 for x in range(5))
print(G)输出:<generator object ....>,3次next(G),next(G),next(G)后,输出0,2,4
方法二:②函数中用yield返回
每次调用next函数,会执行生成器代码,一直到yield返回一个数据。下次调用next函数,会从当前停止的yield之后开始执行。
(1)函数式生成器的简单形式
def G(): i = 0 yield i i += 1 yield i i += 1 yield i g1=G() print(next(g1)) print(next(g1)) print(next(g1))
说明:
使用了yield后,就不再是函数,而是生成器。
yield的作用有2点:一是记录当前运行状态(断点),二是返回值,类似return。
next()函数让生成器从断点出继续执行,即唤醒生成器。
(2)函数式生成器的循环形式
def G2(): i = 0 while i < 5: yield i # 下次调用next函数,会从yield之后开始执行。 i += 1 g1 = G2() print(next(g1)) # 每次调用next函数,会执行生成器的代码,一直到yield返回一个数据。 print(next(g1)) print(next(g1))
输出:
0
1
2
(3)函数式迭代器 - 斐波拉契数列的生成器
def fib_generator(n): '''斐波那契数列生成器 :param n: 可遍历(迭代)的次数 :return: yield ''' current = 0 # 标记迭代器位置 # 初始值 num1 = 0 num2 = 1 while current < n: num = num1 num1, num2 = num2, num1 + num2 current += 1 yield num def main(): f = fib_generator(10) for x in f: print(x, end=", ") if __name__ == '__main__': main()
输出:
0, 1, 1, 2, 3, 5, 8, 13, 21, 34,