生成器函数

Posted 2021-01-16 balcklightning

什么是生成器，生成器实质是迭代器，

在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