生成器

Posted ipython-201806

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了生成器相关的知识,希望对你有一定的参考价值。

一、简介

  • 生成器本质就是迭代器
  • 生成器对象可以直接进行for循环

1. 生成器特点

  • 省内存
  • 惰性机制
  • 只能向前,不能反复

2. 获取生成器获取方式

  • 通过生成器函数
  • 通过生成器表达式

二、生成器函数

1. yield

  • 可以把函数分段运行
  • 作用和return一样,也是返回数据

2. 普通函数与生成器函数区别

普通函数:

def func():
    print("111")
    return 222

ret = func()
print(ret)


结果:
111
222

生成器函数:

def func():
    print("111")
    yield 222

ret = func()
print(ret)



结果:
<generator object func at 0x0000018007451C50>

所以:

  • 当函数中存在yield,那么这个函数就是一个生成器函数
  • 当执行生成器函数时,实则为获取这个生成器

3. 获取生成器

def func():
    print("111")
    yield 222

ret = func()  # 获取到生成器

4. 执行生成器

  • 通过__next__()
def func():
    print("111")
    yield 222

gener = func()
ret = gener.__next__()  # 打印 111,222返回给ret
print(ret)

当程序运行完最后一个yield,那么后面继续执行__next__(),程序会报错,但后面内容还会执行。

def func():
    print("111")
    yield 222
    print("333")
    yield 444
    print("555")

gener = func()
print(gener.__next__())  # 111 222
print(gener.__next__())  # 333 444
print(gener.__next__())  # 555 StopIteration

 

  • 通过send()

    send()和__next__()一样都可以让生成器执行到下一个yield,但send()可以给上一个yield的位置变量传递值。当执行完最后一个yield,再继续执行send()时,程序报错,但还可给最后一个yield位置变量传递值。

def func():
    print("111")
    a = yield 222
    print("a = ", a)
    print("333")
    b = yield 444
    print("b = ", b)

gener = func()
gener.__next__()  # 111
print(gener.send("1"))  # a = 1 333 444
print(gener.send("2"))  # b = 2  StopIteration

 

 send和__next__()区别:

  • send和__next__()都是让生成器向下走一次
  • send可以给上一个yield的位置传递值,不能给最后一个yield发送值。在第一次执行生成器代码的时候不能使用send()

 5. 利用for循环获取生成器内部元素

def func():
    yield 111
    yield 222
    yield 333
    yield 444

gen = func()
for i in gen:
    print(i)


结果:
111
222
333
444

6. yield from

    可以直接把可迭代对象中的每一个数据作为生成器的结果进行返回

def gen():
    lst = [11, 22, 33, 44, 55, 66]
    yield from lst


ret = gen()
for i in ret:
    print(i)


结果:
11
22
33
44
55
66

 

此时,上述代码相当于:

def gen():
    lst = [11, 22, 33, 44, 55, 66]
    yield lst[0]
    yield lst[1]
    yield lst[2]
    yield lst[3]
    yield lst[4]
    yield lst[5]


ret = gen()
for i in ret:
    print(i)

 

三、生成器表达式

1. 写法

  • (结果 for 变量 in 可迭代对象 if 条件筛选)
gen = (i for i in range(10))
print(gen)  # <generator object <genexpr> at 0x000002CE52D91C50>

 

 2. 特点

  • 本质是迭代器
  • 省内存
  • 惰性机制
  • 只能向前,不能反复

3. 应用

1)list ()可把传递进来的数据转化成列表,list里面包含for循环

g = (i for i in range(10))

print(list(g))  # [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

2)

def func():
    print(111)
    yield 222

g = func()
g1 = (i for i in g)
g2 = (i for i in g1)

print(list(g)) # 111 [222]
print(list(g1)) # []
print(list(g2)) # []

 

以上是关于生成器的主要内容,如果未能解决你的问题,请参考以下文章

Apollo Codegen 没有找到生成代码的操作或片段

前端开发工具vscode如何快速生成代码片段

前端开发工具vscode如何快速生成代码片段

vscode代码片段生成vue模板

VS Code配置snippets代码片段快速生成html模板,提高前端编写效率

vs 2010代码片段