迭代器和生成器函数

Posted 2020-09-30 黎明NB

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了迭代器和生成器函数相关的知识，希望对你有一定的参考价值。

引入：

l = [1,2,3,4,5]
s = {1,2,3,4}
for i in l:
    print(i)

如果代码是：

for i in 50:
    print(i)

这个运行不了。输出结果是 ‘int‘ object is not iterable

iterable是可迭代的意思。

哪些可以迭代呢？这些可以str、list、tuple、set、dic等

可迭代的标志是 _iter_。

那我们如何判断是否可以迭代呢？

print(‘__iter__‘ in dir([1,2,3]))

双下划线的内置方法一般不用这种方法调用：

print(next(l.__iter__()))

而是用这种方法调用：

print(next(iter(l)) )

小结：

（1）可迭代协议——凡是可迭代的内部都有一个__iter__方法

（2）迭代器里既有iter方法，又有next方法 ——迭代器协议

（3）通过iter(o)得到的结果就是一个迭代器

（4）0是可迭代的对象

判断是否是迭代器和可迭代对象的方法：

from collections import Iterable
from collections import Iterator
s = ‘abc‘
print(isinstance(s,Iterable))
print(isinstance(s,Iterator))
print(isinstance(iter(s),Iterator))

生成器包括两个方面：生成器函数和生成器表达式

生成器有两种：一种是调用方法直接返回的，一种是可迭代对象通过执行iter方法得到的，迭代器有的好处是可以节省内存。

如果在某些情况下，我们也需要节省内存,就只能自己写。我们自己写的这个能实现迭代器功能的东西就叫生成器。

python中内置的生产器：

（1）生成器函数：关键字是yield，yield语句一次返回一个结果，在每个结果中间，挂起函数的状态，以便下次继续运行。

（2）生成器表达式：类似于列表推导式，但是返回的是一个对象而不是一个结果列表，即惰性运算。

生成器函数：一个含有yield的函数就是生产器函数。通过调用返回的是一个对象，而不是结果列表。通过每一次对这个对象的请求就得到一个值def func(): print(‘aaaa‘)

    a=1
    yield a#返回第一个值
    print(‘bbbbb‘)
    yield 12#返回第二个值
    print(‘cccccc‘)
    yield 56#返回第三个值
ret=func()#拿到一个生成器
print(next(ret))#取第一个值
print(next(ret))#ret不可以用func替代
print(next(ret))

生成器例子：可以监听一个文本输入的文本

import time
def tail(filename):
    with open(filename) as f:
        f.seek(0,2)
        while True:
            line=f.readline()
            if not line:
                #time.sleep()
                continue
            yield line
for line in tail(‘监视文件‘):
    print(line,end=‘‘)

可以计算移动平均值的例子:

def averager():
    total=0
    day=0
    average=0
    while True:
        day_num=yield average
        total+=day_num
        day+=1
        average=total/day
avg=averager()#直接返回生成器
next(avg)#激活生成器avg.sed（），什么都 不send和next效果一样
print(avg.send(10))
print(avg.send(15))
print(avg.send(80))

装饰器版计算移动平均值的例子：

def wrapper(func):
    def inner(*arge,**kwargs):
        ret=func(*arge,**kwargs)
        next(ret)
        return ret
    return inner
@wrapper
def averager():
    total=0
    day=0
    average=0
    while True:
        day_num=yield average
        total+=day_num
        day+=1
        average=total/day
avg=averager()#直接返回生成器
# next(avg)#激活生成器avg.sed（），什么都 不send和next效果一样
print(avg.send(10))
print(avg.send(15))
print(avg.send(80))

列表推导式和生成器表达式：

列表推导式：（两个代码实现相同的效果）

for i in range(100):
    print(i*i)
l=[i*i for i in range(100)]
print(l)

l = [{‘name‘:‘v‘,‘age‘:28},{‘name‘:‘v‘}]
name_list = [dic[‘name‘] for dic in l]
print(name_list)

生成器表达式：

l=(i*i for i in range(100))
print(next(l))
print(next(l))
print(next(l))
print(next(l))

l = [{‘name‘:‘v1‘,‘age‘:28},{‘name‘:‘v2‘}]
name_list_generator = (dic[‘name‘] for dic in l)
#print(name_list_generator)
print(next(name_list_generator))
print(next(name_list_generator))