一文掌握python三大神器（迭代器、生成器、装饰器）（一）

目录

一、迭代器 (Iterator)

1、iter创建迭代器

2、自定义迭代器

二、生成器 (Generator)

1、定义生成器函数

2、生成器表达式

3、生成器是特殊的迭代器，迭代器不一定是生成器。

        1）列表迭代器：

        2）字符串迭代器：

        3）字典键迭代器：

        4）集合迭代器

三、装饰器 (Decorator)

1、函数装饰器

2、多个装饰器的使用

3、参数化装饰器

4、类装饰器

1）计数器装饰器

2）计时装饰器（性能监控）

3）除了在__init__中接收函数外，还可以在__call__中直接接收函数

四、总结

---

一、迭代器 (Iterator)

迭代器是Python中用于遍历集合元素（如列表、字典、字符串等）的一种接口，它提供了访问集合元素的方式，而不需要知道集合的内部结构。迭代器有两个基本方法（实现了迭代器协议）：__iter__() 和 __next__()

• __iter__() 方法返回迭代器对象本身，以便你可以使用 for...in... 循环。
• __next__() 方法返回集合的下一个项目，当没有更多项目时，会抛出 StopIteration 异常。

my_list = [1, 2, 3]
it = iter(my_list)
print(next(it))  # 输出: 1
print(next(it))  # 输出: 2
print(next(it))  # 输出: 3

# 为了避免StopIteration异常，可以给next()提供一个默认值作为第二个参数，
# 这样当迭代器耗尽时，会返回这个默认值而不是抛出异常
print(next(it, "No more numbers"))  # 输出: No more numbers

print(next(it))  # 输出: StopIteration

1、iter创建迭代器

iter() 是Python中的一个内置函数，用于从可迭代对象（如列表、元组、字符串、字典、集合或生成器等）创建一个迭代器 (即将一个可迭代对象转换为迭代器)。迭代器是访问集合元素的一种方式，提供了对集合元素进行遍历访问的接口，且只允许向前移动。

当调用iter()函数时，需要传入一个可迭代对象作为参数。函数返回一个迭代器对象，你可以通过这个迭代器对象使用next()函数（或在for循环中隐式地使用）来逐个访问原可迭代对象中的元素。

# example 1, list
my_list = [1, 2, 3]
my_iterator = iter(my_list)
print(next(my_iterator))  # 输出: 1
print(next(my_iterator))  # 输出: 2


# example 2, str
my_string = "hello"
my_iterator = iter(my_string)
print(next(my_iterator))  # 输出: 'h'
print(next(my_iterator))  # 输出: 'e'

2、自定义迭代器

可以为自定义类定义__iter__和__next__方法来创建一个可迭代对象，然后使用iter()函数转换为迭代器。

class CountDown:
    def __init__(self, start):
        self.current = start
    
    def __iter__(self):
        return self
    
    def __next__(self):
        if self.current <= 0:
            raise StopIteration
        else:
            # 这里修改了计数器的状态， 下面的return语句会返回修改后的状态，
            # 例如第一次self.current等于3，减1后等于2, return 2 + 1 = 3, 
            # 第二次self.current减1后等于1, return 1 + 1 = 2, 依次类推, 实现倒计时功能
            self.current -= 1
            return self.current + 1

countdown = CountDown(3)
iterator = iter(countdown)
print(type(iterator))  # 输出: <class '__main__.CountDown'>
print(next(iterator))  # 输出: 3
print(next(iterator))  # 输出: 2
print(next(iterator))  # 输出: 1

二、生成器 (Generator)

生成器是一种特殊的迭代器，使用 yield 语句而不是 return 语句来产生一系列的值。生成器在执行到 yield 时会暂停，并记住当前状态，下次从同一位置继续执行。生成器可以节省内存，特别适合处理大量数据流或生成大数据序列。

1、定义生成器函数

def simple_generator():
    yield 1
    yield 2
    yield 3

gen = simple_generator()
print(type(gen))  # <class 'generator'>
print(next(gen))  # 输出: 1
print(next(gen))  # 输出: 2

2、生成器表达式

列表推导式的生成器版本，使用圆括号而不是方括号包围。生成器表达式在内存使用上更为高效，特别是在处理大量数据时。

例如，以下是一个生成器表达式的例子，生成一个平方数序列：

squares_gen = (x**2 for x in range(10))
print(type(squares_gen))  # <class 'generator'>
for square in squares_gen:
    print(square)

3、生成器是特殊的迭代器，迭代器不一定是生成