生成器和推导式

生成器定义：

函数中带yield的就是生成器

def foo():
    print(1)
    yield 4
    print(2)
    yield 5
    print(3)
    yield 6               # 每次执行到yield,没有则报错
g=foo()             # 产生一个生成器
print(g.__next__())
print(g.__next__())
print(g.__next__())

 

eg：

def gen():
    for i in range(200):
        yield i
g=gen()
for i in range(10):
    print(g.__next__())

生成器本质就是迭代器，拥有惰性机制，每执行__next__()或者next()才会向下执行取值

 

eg：

def foo():
    lst=[1,2,3,5]
    yield from lst
g=foo()
print(next(g))
print(next(g))
print(next(g))
print(next(g))

大坑: 生成器函数运行之后. 产生一个生成器. 而不是运行函数

def func():
    print("我叫周润发")
    yield "林志玲"   # yield表示返回. 不会终止函数的执行
    print("宝宝干嘛去了??")
    yield "宝宝回来了"
    print("宝宝你在干嘛?")
    # yield "没了"

ret = func() # 执行函数, 此时没有运行函数.

此时拿到的是生成器

print("返回值是", ret) # <generator生成器 object func at 0x0000000009E573B8>

send() -> __next__()
send()可以给上一个yield位置传值

def func():
    print("韭菜盒子")
    a = yield "韭菜鸡蛋"
    print("a", a)
    b = yield "韭菜西红柿"
    print("b", b)
    c = yield "火烧"
    print("c", c)

gen = func()
print(gen.__next__()) # 第一个位置用send没有任何意义,报错，除非 gen.send(None)
print(gen.send("篮球")) # 给上一个yield位置传值
print(gen.send("足球"))

 

eg：

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__()
print(ret1)
ret2 = gen.send("胡辣汤")
print(ret2)
ret3 = gen.send("狗粮")
print(ret3)
ret4 = gen.send("猫粮")
print(ret4)

 

列表推导式：

语法：

[结果 语句 条件]

eg:打印1-100偶数的平方

lst=[ i**2 for i in range(1,101) if i%2==0]
print(lst)

　eg：打印名字中带两个e的名字

names=[['tom','meiko','jeffer','wesley'],['alice','jill','anee','haeeae']]
lst=[name for el in names for name in el if name.count("e")==2]
print(lst)

 

字典推导式：

语法：{结果(key:value)for循环 if条件}

lst=[11,22,33] # {0:11,1:22,2:33}
dic={i:lst[i] for i in range(len(lst))}
print(dic)

 

集合推导式：

语法：{k for 循环 if 条件}

dic={'1':2,'2':'6'}
d={i for i in dic.items()}
print(d)

生成器表达式：

g=(i for i in range(10))    
print(g)                      #<generator object <genexpr> at 0x0000024294CAF4C0>
print(g.__next__()) #0
print(g.__next__())
print(g.__next__())
print(g.__next__())
print(g.__next__())
print(g.__next__())
print(g.__next__())
print(g.__next__())
print(g.__next__())
print(g.__next__())#9