Python3 迭代器与生成器

最新推荐文章于 2024-06-19 19:22:38 发布

Z_凌云

最新推荐文章于 2024-06-19 19:22:38 发布

阅读量104

点赞数

分类专栏： Python 文章标签： python 开发语言

本文链接：https://blog.csdn.net/m0_46983541/article/details/121247879

版权

Python 专栏收录该内容

22 篇文章 7 订阅

订阅专栏

本文介绍了Python迭代器的基础概念、基本操作，包括iter()和next()方法，演示了如何创建和使用迭代器，并重点讲解了生成器的概念及实际应用，如斐波那契数列的生成。通过实例学习，理解迭代器在数据处理中的关键作用。

摘要由CSDN通过智能技术生成

一、迭代器介绍
迭代是Python最强大的功能之一，是访问集合元素的一种方式，它可以记住遍历的位置的对象。迭代器对象从集合的第一个元素开始访问，直到所有的元素被访问完结束。迭代器只能往前不会后退。
二、迭代器的基本方法
迭代器有两个基本的方法：iter() 和 next()。
（1）使用实例：

list=[12,13,15,16]  #创建一个列表
T = iter(list)      #创建迭代对象
print(next(T))      #输出迭代器下一个元素

print(next(T))      #继续输出下一个元素

输出结果：
在这里插入图片描述
（2）迭代器对象用for语句进行遍历：

list=[12,13,15,16]
it = iter(list)    # 创建迭代器对象
for x in it:
    print (x, end=" ")

输出结果：
在这里插入图片描述
（3）使用next函数：

import sys  # 引入 sys 模块

list = [12,13,15,16]
it = iter(list)  # 创建迭代器对象
while True:
    try:
        print(next(it))
    except StopIteration:
        sys.exit()

输出结果：
在这里插入图片描述
三、创建一个迭代器
把一个类作为一个迭代器使用需要在类中实现两个方法 iter() 与 next() 。在面向对象编程里，类都有一个构造函数，Python 的构造函数为 init(), 它会在对象初始化的时候执行。
iter() 方法返回一个特殊的迭代器对象，这个迭代器对象实现了 next() 方法并通StopIteration 异常标识迭代的完成。
next() 方法会返回下一个迭代器对象。
实例一：创建一个返回数字的迭代器，初始值为 1，逐步递增 1。

class MyNumbers:
    def __iter__(self):
        self.a = 1
        return self

    def __next__(self):
        x = self.a
        self.a += 1
        return x

myclass = MyNumbers()
myiter = iter(myclass)

print(next(myiter))
print(next(myiter))
print(next(myiter))
print(next(myiter))
print(next(myiter))

输出结果：
在这里插入图片描述
实例二：StopIteration 异常用于标识迭代的完成，防止出现无限循环的情况，在 next() 方法中我们可以设置在完成指定循环次数后触发 StopIteration 异常来结束迭代。
在 20 次迭代后停止执行：

class MyNumbers:
    def __iter__(self):
        self.a = 1
        return self

    def __next__(self):
        if self.a <= 20:
            x = self.a
            self.a += 1
            return x
        else:
            raise StopIteration

myclass = MyNumbers()
myiter = iter(myclass)

for x in myiter:
    print(x)

输出结果：
在这里插入图片描述
四、生成器
在 Python 中，使用了 yield 的函数被称为生成器（generator）。跟普通函数不同的是，生成器是一个返回迭代器的函数，只能用于迭代操作。在调用生成器运行的过程中，每次遇到 yield 时函数会暂停并保存当前所有的运行信息，返回 yield 的值, 并在下一次执行 next() 方法时从当前位置继续运行。调用一个生成器函数，返回的是一个迭代器对象。
使用 yield 实现斐波那契数列：

def fibonacci(n):  # 生成器函数 - 斐波那契
    a, b, counter = 0, 1, 0
    while True:
        if (counter > n):
            return
        yield a
        a, b = b, a + b
        counter += 1

f = fibonacci(10)  # f 是一个迭代器，由生成器返回生成

while True:
    try:
        print(next(f), end=" ")
    except StopIteration:
        sys.exit()