python 之 函数 生成器

5.10 生成器

函数内有yield关键字，再调用函数就不会立刻执行函数体代码，会得到一个返回值，该返回值就是生成器，生成器本质就是迭代器

def chicken():

    print('=====>first')

    yield 1

    print('=====>sencond')

    yield 2

    print('=====>third')

    yield 3

obj=chicken()

#生成器本质就是迭代器，也就是说生成器的玩法其实就是迭代器的玩法

print(obj.__iter__() is obj)

res=obj.__next__()

print(res)

​

res1=obj.__next__()

print(res1)

​

res2=obj.__next__()

print(res2)

1、iter_obj=obj.iter()，拿到迭代器 2、出发iter_obj.next()，拿到该方法的返回值，赋值给item 3、周而复始，直到函数内不在有yield，即取值完毕 4、for会检测到StopIteration异常，结束循环

yield功能： 1、提供了一种自定义迭代器的方式 2、yield可以返回多次值

def chicken():

    print('=====>first')

    yield 1

    print('=====>sencond')

    yield 2

    print('=====>third')

    yield 3

​

obj=chicken()

for item in obj:

     print(item)

range()

def my_range(start,stop,step=1):

    n=start

    while n < stop:

        yield n #yield 4

        n+=step #

obj=my_range(3,7,2) #3,5,

print(obj.__next__())

print(obj.__next__())

print(obj.__next__())

print(obj.__next__())

print(obj.__next__())

def my_range(start,stop,step=1):

    n=start

    while n < stop:

        yield n #yield 5

        n+=step #

for item in my_range(5,10,2):

    print(item)

5.10.1 生成器之yield的表达式形式

def eat(name):

    print('%s ready to eat' %name)

    food_list=[]

    while True:

        food=yield food_list # food='骨头'

        food_list.append(food) #food_list=['泔水','骨头']

        print('%s start to eat %s' %(name,food))

dog1=eat('alex')

1、必须初始化一次，让函数停在yield的位置

res0=dog1.__next__()

print(res0)

接下来的事，就是喂狗,send有两方面的功能

1、给yield传值

2、同__next__的功能

res1=dog1.send('泔水')

print(res1)

res2=dog1.send('骨头')

print(res2)

res3=dog1.send('shit')

print(res3)