函数进阶--生成器与迭代器

Posted 2020-11-06 游走在边缘的人

列表生成式

列表生成式就是将一些简单的代码合并成一条代码来写

# 需要将列表[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]各自加1
#普通会这样写
a = [1,3,4,6,7,7,8,9,11]
for index,i in enumerate(a):
    a[index] +=1
print(a)
#用列表生成式
a = [i+1 for i in range(10)]
print(a)

生成器

带有 yield 关键字的的函数在 Python 中被称之为 generator(生成器)。Python 解释器会将带有 yield 关键字的函数视为一个 generator 来处理。一个函数或者子程序都只能 return 一次，但是一个生成器能暂停执行并返回一个中间的结果 —— 这就是 yield 语句的功能 : 返回一个中间值给调用者并暂停执行。

生成器有两种创建方式：

• 列表生成式,以（）形式
• 函数

yield、return、next

• yield 返回数据，并冻结当前的执行过程
• return 返回数据，并结束当前函数
• next 是唤醒冻结函数的执行过程，继续执行，直到遇到下一个yield
• 当函数里有了yield之后，函数名加（）就可以变成生成器
• return在生成器里代表生成器的中止，会直接报错

通过生成器来实现日志文件

通过生成器写一个日志调用方法， 支持以下功能
根据指令向屏幕输出日志
根据指令向文件输出日志
根据指令同时向文件&屏幕输出日志
以上日志格式如下
代码：

import time
cont = 0
def logger(filename,channel=‘file‘):
    """
    日志方法
    :param filename: log filename
    :param channel: 输出的目的地，屏幕(terminal)，文件(file)，屏幕+文件(both)
    :return:
    """
    global cont
    while True:
        cont += 1
        msg = yield
        a = time.strftime(‘%Y-%m-%d %X‘)
        info = ‘%s[%s]%s‘%(a,cont,msg)
        if channel == ‘file‘:
            with open(filename,‘a‘,encoding=‘utf-8‘) as f:
                f.write(‘\n%s‘%info)
        elif channel == ‘terminal‘:
            print(info)
        elif channel == ‘both‘:
            with open(filename,‘a‘,encoding=‘utf-8‘) as f:
                f.write(‘\n%s‘%info)
            print(info)
a = logger(filename=‘a.txt‘,channel=‘both‘)
next(a)
a.send(‘user alex login success‘)

迭代器

判断一个对象是否是可迭代对像可以用isinstance()来进行判断

>>> from collections import Iterable
>>> isinstance([], Iterable)
True
>>> isinstance({}, Iterable)
True
>>> isinstance(‘abc‘, Iterable)
True
>>> isinstance((x for x in range(10)), Iterable)
True
>>> isinstance(100, Iterable)
False

可以被next()函数调用并不断返回下一个值的对象称为迭代器：Iterator。
可以使用isinstance()判断一个对象是否是Iterator（迭代器）对象

>>> from collections import Iterator
>>> isinstance((x for x in range(10)), Iterator)
True
>>> isinstance([], Iterator)
False
>>> isinstance({}, Iterator)
False
>>> isinstance(‘abc‘, Iterator)
False

生成器都是Iterator对象，但list、dict、str虽然是Iterable，却不是Iterator。
把list、dict、str等Iterable变成Iterator可以使用iter()函数：

>>> isinstance(iter([]), Iterator)
True
>>> isinstance(iter(‘abc‘), Iterator)
True

• 凡是可作用于for循环的对象都是Iterable(可迭代对象)类型；
• 凡是可作用于next()函数的对象都是Iterator类型，它们表示一个惰性计算的序列；
• 集合数据类型如list、dict、str等是Iterable但不是Iterator，不过可以通过iter()函数获得一个Iterator对象。