python基础day-12：python三神器（装饰器，迭代器，生成器）、模块

本文链接：https://blog.csdn.net/weixin_41615044/article/details/111593569

三神器

一、装饰器

1.1 无参装饰器

无参装饰器

1.2 有参装饰器

应用场景
- 实现装饰器功能的时候，需要额外的数据

def 函数名（参）：
    无参装饰器 - 里面会用到额外参数
    return 无参装饰器函数名

函数名 - 创建装饰器的函数名
参数列表 - 有多个参数看实现装饰器功能额外需要多少数据

# 写个装饰器将原函数返回值加上指定的数
def add_number(num):
    def add_number_(fn):
        def new_fn(*args, **kwargs):
            result = fn(*args, **kwargs)
            if type(result) in (int, float, complex):
                return result + num
            return result
        return new_fn
    return add_number_


@add_number(5)
def func1(num1, num2):
    return num1 * num2


print(func1(2, 4))

二、迭代器（iter）

2.1 什么是迭代器

容器型数据类型
不可直接获取所有元素，只能一个一个取走
不可直接统计元素个数
创建：
1. 将其它序列转换成迭代器
2. 生成器创建

list1 = [1, 34, 5, 2, 456, 35]
iter1 = iter(list1)

print(iter1, type(iter1))  # <list_iterator object at 0x0000022A37920BC8> <class 'list_iterator'>

2.2 获取迭代器元素

获取到迭代器中的元素，该元素在迭代器中就不存在了

2.2.1 单个

next(迭代器)
没有了就会报错

print(next(iter1))  # 1
print(next(iter1))  # 34

2.2.2 遍历

for 变量 in 迭代器：
遍历后，迭代器也会空

iter2 = iter(range(5))
for x in iter2:
    print(x)  # 0 1 2 3 4
print(list(iter2))  # []

iter3 = iter(range(0, 10, 2))
list(iter3)
# print(next(iter3))  # StopIteration

三、生成器（generator）

3.1 什么是生成器

特殊的迭代器
存储的是产生数据的算法而不是数据本身
是个下蛋的母鸡
内存消耗少

3.2 创建生成器

调用带有yield关键字的函数就可以得到
不管yield在函数的哪里，哪怕是放在return后面，调用时都不会执行函数体
函数调用的值是一个生成器对象

def func1():
    yield

3.3 生成器产生数据的能力怎么确定

创建多少数据？
1. 执行完生成器对应的函数体会遇到几次yield，就创造多少个数据
2. 每次遇到yield，yield后面的值是什么，对应创造的数据就是什么

def func2():
    for i in range(5):
        yield 10**i


print(next(func2()), next(func2()))  # 1 1

gen1 = func2()
print(next(gen1), next(gen1))  # 1 10

3.4 生成器产生数据的原理

生成器对应的函数在调用函数的时候不会执行函数体
获取生成器中的数据时才会执行函数体
每次获取数据时，从上一次结束的位置开始执行函数体
每次遇到yield就停下来，将yield后面的数据作为结果返回，并记录结束位置
如果next去取，到函数结束都没有遇到yield，就会报错

# 练习：产生指定前缀和长度的学号
# 前缀：py，长度：4  py0001~py9999
def prefix_number(prefix, width):
    for i in range(1, 10**width):
        yield prefix + str(i).zfill(width)


pn_gen = prefix_number('py', 4)
for i in range(10):
    print(next(pn_gen))