Java转python第五天

Posted 2021-11-30 wgsn

一、简单生成器函数

与函数不同的是返回不用return，而是用yield。生成器函数返回的是一个迭代器。

def func():
    print(1)
    yield 2
    print(3)
    yield 4
a =func()
for i in a:
    print(i)
a =func()
print(a.__next__())
print(a.__next__())

例子：要求生产100000个包子，但是又不可能一次性吃完，每次调用eat()函数时，传入要吃的数量，在进行生产，

　　不会造成控件的浪费

def produce():
    for i in range(1,100001):
        yield i
a = produce()
def eat(num):
    for i in range(num):
        print(a.__next__())
eat(10)
eat(20)

例子