迭代器与生成器

迭代器

 

迭代取值和索引取值的对比：

迭代器主要就是一个迭代取值，另外一种取值方式就是索引(下标)取值

l = [1, 2, 3, 4]

res = l.__iter__()
res.__next__() # 1
res.__next__() # 2

res1 = l.__iter__()
res1.__next__()  # 1
res1.__next__()  # 2

 迭代取值

    1. 不依赖于索引取值的一种取值方式
    2. 不能够重复取值，只能够从左往右固定取值

索引取值

    1. 它能够重复取值(通过索引l[0])
    2. 它需要是容器类型 

生成器

一 生成器与yield

若函数体包含yield关键字，再调用函数，并不会执行函数体代码，得到的返回值即生成器对象

def my_range(start,stop,step=1):
    print('start...')
    while start < stop:
        yield start
        start+=step
    print('end...')

g=my_range(0,3)
# g
# <generator object my_range at 0x104105678>

生成器内置有__iter__和__next__方法，所以生成器本身就是一个迭代器

g.__iter__
# <method-wrapper '__iter__' of generator object at 0x1037d2af0>

g.__next__
# <method-wrapper '__next__' of generator object at 0x1037d2af0>

 因而我们可以用next(生成器)触发生成器所对应函数的执行，

next(g) # 触发函数执行直到遇到yield则停止,将yield后的值返回，并在当前位置挂起函数
start...
# 0
next(g) # 再次调用next(g)，函数从上次暂停的位置继续执行，直到重新遇到yield...
# 1
next(g) # 周而复始...
# 2
next(g) # 触发函数执行没有遇到yield则无值返回，即取值完毕抛出异常结束迭代
end...

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
StopIteration

 既然生成器对象属于迭代器，那么必然可以使用for循环迭代，如下：

for i in countdown(3):
    print(i)

'''
countdown start
3
2
1
Done!
'''

 有了yield关键字，我们就有了一种自定义迭代器的实现方式。yield可以用于返回值，但不同于return，函数一旦遇到return就结束了，而yield可以保存函数的运行状态挂起函数，用来返回多次值

二 yield表达式应用

在函数内可以采用表达式形式的yield

def eater():
    print('Ready to eat')
    while True:
        food=yield
        print('get the food: %s, and start to eat' %food)

可以拿到函数的生成器对象持续为函数体send值，如下

g=eater() # 得到生成器对象
# g
# <generator object eater at 0x101b6e2b0>

next(e) # 需要事先”初始化”一次，让函数挂起在food=yield，等待调用g.send()方法为其传值
# Ready to eat

g.send('包子')
# get the food: 包子, and start to eat
    
g.send('鸡腿')
# get the food: 鸡腿, and start to eat

针对表达式形式的yield，生成器对象必须事先被初始化一次，让函数挂起在food=yield的位置，等待调用g.send()方法为函数体传值，g.send(None)等同于next(g)。

我们可以编写装饰器来完成为所有表达式形式yield对应生成器的初始化操作，如下

def init(func):
    def wrapper(*args,**kwargs):
        g=func(*args,**kwargs)
        next(g)
        return g
    return wrapper

@init
def eater():
    print('Ready to eat')
    while True:
        food=yield
        print('get the food: %s, and start to eat' %food)

表达式形式的yield也可以用于返回多次值，即变量名=yield 值的形式，如下

def eater():
    print('Ready to eat')
    food_list=[]
    while True:
        food=yield food_list
        food_list.append(food)

e=eater()
next(e)
# Ready to eat
# []
e.send('蒸羊羔')
# ['蒸羊羔']
e.send('蒸熊掌')
# ['蒸羊羔', '蒸熊掌']
e.send('蒸鹿尾儿')
# ['蒸羊羔', '蒸熊掌', '蒸鹿尾儿']

三 yield和return的对比

yield

1. 代码遇到yield不会停止，而是停住
2. yield也可以有返回值，并且还支持多个，以元组的形式返回
3. yield可以把一个函数变成生成器，next取值

 

return 

1. 代码遇到return就会停止
2. return可以有返回值并且还支持多个，以元组的形式返回