迭代器与生成器#

1
注意区分
2
list/tuple/dict = 可迭代对象（Iterable）→ 能造迭代器
3
generator = 迭代器（Iterator）→ 本身就是工人
4
可迭代对象：能被 for 循环的东西（有 __iter__ 方法）
5
迭代器：能被 next() 调用的东西（有 __iter__ + __next__ 方法）
6
所有迭代器都是可迭代对象，但反之不成立
7

8
迭代器协议 = __iter__ + __next__
9
__iter__：返回迭代器自身（或另一个迭代器）
10
__next__：返回下一个值，或抛 StopIteration
11
一个合格的迭代器必须要实现上面的两个方法，否则py可不认哦🤡

1
class Fatchicken:
2
    def __init__(self,egg):
3
        self.egg = egg
4
    def __iter__(self):
5
        return self#返回迭代器本身(必须实现！)
6
    def __next__(self):
7
        if self.egg<=0:
8
            raise StopIteration #告诉for循环该结束了！
9
        self.egg -=1
10
        return self.egg+1#return会导致函数结束，所以__next__函数里先对数据操作再返回
11
aa = [i for i in Fatchicken(3)]
12
print(aa)

1
with open('file.txt') as f:
2
    print(type(f))  # <class '_io.TextIOWrapper'>
3
    print(hasattr(f, '__next__'))  # True → f就是迭代器
4
for line in f:  # 逐行读，不加载全文件
5
    pass

1
#手搓生成器
2
def eat(a):
3
    while a>0:
4
        yield a
5
        a -=1
6
bb=[i for i in eat(3)]
7
print(bb)

1
yield 的魔法：
2
函数遇到 yield → 暂停执行，保存状态
3
下次调用 __next__() → 从暂停处继续
4
自动生成：
5
__iter__()
6
__next__()
7
StopIteration 处理