生成器python

生成器推导式

g=(x*2 for x in range(10)) #生成器推导式
while True:
    try:
        e=next(g)
        print(e)
    except:
        print("没有更多的元素了!")
        break

定义生成器的方式二借助函数完成

只要函数中出现yield关键字，函数就变成了迭代器
步骤：
1.定义一个函数，函数中使用yield关键字
2.调用函数接受调用的结果
3.得到的结果就是生成器
4.借助next()和__next__得到元素
yield n:表示返回这个n,并将程序暂停在这里。

def func():
    n=0
    while True:
        n+=1
        yield n #return n 暂停
g=func()
print(g) #<generator object func at 0x000001DAE5037920>
print(next(g))
print(next(g))
print(next(g))

斐波那契数列 0,1,1,2,3,5,8

def fib(length):
    a,b=0,1
    n=0
    while n<length:
        #print(b)
        yield b
        a,b=b,a+b
        n+=1
    return "没有更多元素了！！"
g=fib(8)
print(next(g))
print(g.__next__())
print(next(g))
print(next(g))

生成器方法：

__next__() :获取下一个元素
send(value): 向每次生成器调用中传值：注意：第一次调用要send(None)

def gen():
    i=0
    while i<5:
        temp=yield i
        print("temp",temp)
        i+=1
    return "没有更多的数据"
g=gen()
print(g.send(None))
n1=g.send('哈哈')
print("n1:",n1)
n2=g.send('嘿嘿')
print("n2:",n2)

生成器应用

交替进行。

#进程》线程》协程
def task1(n):
    for i in range(n):
        print("搬第{}块砖！".format(i))
        yield
def task2(n):
    for i in range(n):
        print("听第{}首歌".format(i))
        yield
g1=task1(10)
g2=task2(5)
while True:
    try:
        next(g1)
        next(g2)
    except:
        pass