什么是生成器，生成器实质是迭代器，
在python有三种方式获取生成器：
1.通过生成器函数
2.通过各种推导式来实现生成器
3.通过数据的转换也可以获取生成器
首先，我们来看一个很简单的函数：
def func():
print("111")
return 222
ret = func()
print(ret)
结果:
111
222
将函数中的return换成yield就是生成器，也就是说有yield的函数就是生成器函数
def func():
print("111")
yield 222
ret = func()
print(ret)
结果:
<generator object func at 0x10567ff68>
运行的结果和上面运行不一样。为什么呢。由于函数中存在了yield.那个这个函数就是一个生成器函数。这个时候。我们再执行这个函数的时候。就不再是函数的执行。而是获取这个生成器。
如何使⽤呢? 想迭代器. 生成器的本质是迭代器. 所以. 我们可以直接执行__next__()来执⾏行以下⽣生成器.
def func():
print("111")
yield 222
gener = func() # 这个时候函数不不会执行. 而是获取到生成器
ret = gener.__next__() # 这个时候函数才会执行. yield的作⽤和return⼀样. 也是返回 数据
print(ret)
结果:
111
222
yield是分段执行的函数，一个__next__执行一段yield
def func():
print("111")
yield 222
print("333")
yield 444
gener = func()
ret = gener.__next__()
print(ret)
ret2 = gener.__next__()
print(ret2)
结果：
111
222
333
444
return是返回值并直接停止执行函数，所以yield不要和return混用
def func():
print("111")
yield 222
return 3 #在这加了一个return
print("333")
yield 444
gener = func()
ret = gener.__next__()
print(ret)
ret2 = gener.__next__()
print(ret2)
结果：
Traceback (most recent call last):
File "/Users/kaixiangchen/PycharmProjects/untitled1/ss.py", line 10, in <module>
ret2 = gener.__next__()
StopIteration: 3 #报错停止迭代器在第三行
111
222
当程序运行完最后一个yield. 那么后面继续进行__next__()程序会报错.
好了生成器说完了. 生成器有什么作用呢? 我们来看这样⼀个需求. 老男孩向JACK JONES订 购10000套学⽣生服. JACK JONES就比较实在. 直接造出来10000套衣服.
def cloth():
lst = []
for i in range(0, 10000):
lst.append("衣服" + str(i))
return lst
cl = cloth()
但是呢, 问题来了. 老男孩现在没有这么多学生啊. 一次性给我这么多. 我往哪里放啊. 很尴尬 啊. 最好的效果是么样呢? 我要1套. 你给我1套. ⼀一共10000套. 是不是最完美的.
def cloth():
for i in range(0, 10000):
yield "衣服"+str(i)
cl = cloth()
print(cl.__next__())
print(cl.__next__())
print(cl.__next__())
print(cl.__next__())
区别: 第⼀种是直接一次性全部拿出来. 会很占⽤用内存. 第二种使⽤生成器. ⼀次就一个. ⽤多 少生成多少. 生成器是一个一个的指向下一个. 不会回去, __next__()到哪, 指针就指到哪儿.下⼀次继续获取指针指向的值.
接下来我们来看send方法, send和__next__()⼀样都可以让⽣成器执行到下⼀个yield.
send和__next__()区别:
1. send和next()都是让生成器向下走一次
2. send可以给上一个yield的位置传递值,不能给最后⼀个yield发送值.在第一次执行生
成器代码的时候不能使用send()
def eat():
print("我吃什什么啊")
a = yield "馒头"
print("a=",a)
b = yield "⼤大饼"
print("b=",b)
c = yield "⾲韭菜盒⼦子"
print("c=",c)
yield "GAME OVER"
gen = eat() # 获取⽣生成器器
ret1 = gen.__next__() #执行到a = yield "馒头"
print(ret1)
ret2 = gen.send("胡辣汤") #1.先把"胡辣汤"给上一个yield变为a = yield "胡辣汤"2.再从print("a=",a)执行到b = yield "⼤大饼"
print(ret2)
ret3 = gen.send("狗粮")
print(ret3)
ret4 = gen.send("猫粮")
print(ret4)
结果：
我吃什什么啊
馒头
a= 胡辣汤
⼤大饼
b= 狗粮
⾲韭菜盒⼦子
c= 猫粮
GAME OVER
⽣成器可以使用for循环来循环获取内部的元素:
def func():
print(111)
yield 222
print(333)
yield 444
print(555)
yield 666
gen = func()
for i in gen:
print(i)
结果：
111
222
333
444
555
666