怎么自定义一个生成器:
两个方法:
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,