浅析Python中生成器和迭代器的神奇力量

​class MyIterator:
    def __init__(self, data):
        self.data = data
        self.index = 0
​
    def __iter__(self):
        return self
​
    def __next__(self):
        if self.index < len(self.data):
            result = self.data[self.index]
            self.index += 1
            return result
        else:
            raise StopIteration
​
            # 使用自定义的迭代器
my_iter = MyIterator("Hello")
for char in my_iter:
    print(char)

# 列表迭代器
list_iter = iter([1, 2, 3])
print(next(list_iter))
print(next(list_iter))
print(next(list_iter))
​
# 字典迭代器
dict_iter = iter({"a": 1, "b": 2, "c": 3})
print(next(dict_iter))
print(next(dict_iter))
print(next(dict_iter))
​
# 文件迭代器
with open("test.txt") as file:
    for line in file:
        print(line, end="")

​def my_generator(n):
    i = 0
    while i < n:
        yield i
        i += 1
​
        # 使用自定义的生成器
for num in my_generator(5):
    print(num)

​# 创建一个生成器表达式
gen_expr = (i ** 2 for i in range(5))
​
# 使用生成器表达式
for num in gen_expr:
    print(num)