如何在 Python 中使用生成器和迭代器? _Python

Python/ target=_blank class=infotextkey>Python 是一种流行的多功能编程语言，广泛用于各种应用程序，从 Web 开发和数据分析到机器学习和科学计算。该语言最强大的功能之一是它能够与生成器和迭代器一起工作，这提供了一种方便有效的方法来处理各种上下文中的大量数据。

文章插图

在本文中，我们将探讨 Python 中的生成器和迭代器是什么、它们的工作原理以及您可能希望在代码中使用它们的原因。我们还将提供一些简单和复杂的用例来演示这些功能的多功能性。
Python 中的生成器和迭代器是什么？【如何在 Python 中使用生成器和迭代器?】在 Python 中，迭代器是一个可以迭代（循环）的对象，这意味着它可以在 for 循环中使用。迭代器是一个实现了迭代器协议的对象，迭代器协议要求它提供两个方法：iter()和next() 。iter() 方法返回迭代器对象本身，而 next() 方法返回迭代序列中的下一个值。如果没有更多的项目要返回，next() 方法应该引发 StopIteration 异常。
另一方面，生成器是一种特殊类型的迭代器，它是使用函数而不是类定义的。生成器函数是一种包含一个或多个 yield 语句的函数，这些语句会暂时暂停执行并为调用者生成一个值。当再次调用生成器函数时，执行会从中断处恢复，并记住生成器函数的最后状态。这使得动态生成一系列值变得容易，而无需预先计算所有值。
为什么使用生成器和迭代器？生成器和迭代器在各种上下文中都很有用，因为它们提供了一种高效且内存友好的方式来处理大量数据。通过即时生成值或分块迭代大型数据集，您可以避免一次将整个数据集加载到内存中，这对于非常大的数据集来说是不切实际甚至不可能的。
生成器和迭代器对于处理无限或非常大的数据集也很有用，例如来自传感器的流数据或实时处理日志文件。通过在需要用到数据的时候生成或迭代数据，您可以避免一次将所有数据存储在内存中。
生成器和迭代器的用例让我们看一下 Python 中生成器和迭代器的一些简单和复杂的用例：

生成数字序列：生成器最简单的用例之一是生成数字序列。这是一个例子：

def generate_numbers(n):for i in range(n):yield ifor number in generate_numbers(10):print(number)在此示例中，generate_numbers() 函数使用 for 循环和 yield 语句生成从 0 到 n-1 的数字序列。调用该函数时，它会返回一个迭代器，该迭代器可用于 for 循环以即时生成数字。这比使用列表或 range() 函数预先生成整个数字序列更节省内存。

处理大型数据集：生成器和迭代器的另一个常见用例是分块处理大型数据集，而不是一次将整个数据集加载到内存中。这是一个例子：

def process_file(file):with open(file) as f:for line in f:yield line.strip()for line in process_file('data.txt'):print(line)

在此示例中，process_file() 函数从文件中读取大型数据集，并使用 yield 语句逐行生成文件。调用该函数时，它会返回一个迭代器，该迭代器可用于 for 循环以处理从磁盘读取的文件行。这比一次将整个文件读入内存更节省内存，后者对于无法放入内存的非常大的文件可能会产生问题。

过滤值序列：生成器和迭代器也可用于根据特定条件过滤值序列。这是一个例子：

def filter_numbers(numbers):for number in numbers:if number % 2 == 0:yield numbernumbers = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]for even_number in filter_numbers(numbers):print(even_number)

在此示例中，filter_numbers() 函数将数字列表作为输入，并使用 yield 语句和条件语句仅生成偶数。调用该函数时，它会返回一个迭代器，该迭代器可用于 for 循环以仅即时生成偶数。这比使用列表或 filter() 函数预先创建一个新的偶数列表更节省内存。

生成无限序列：生成器也可用于生成无限序列值，例如斐波那契数列。这是一个例子：

在此示例中，fibonacci() 函数使用 while 循环和 yield 语句生成无限的斐波那契数列。调用该函数时，它会返回一个迭代器，该迭代器可用于 for 循环以即时生成斐波那契数列。通过检查每个数字的值并在超过 100 时跳出循环，我们可以只生成我们需要的斐波那契数列，而无需预先计算整个序列。
结论生成器和迭代器是 Python 的强大功能，它们提供了一种方便高效的方式来处理各种上下文中的大量数据。通过即时生成值或分块迭代大型数据集，您可以避免一次将整个数据集加载到内存中，这对于非常大的数据集来说是不切实际甚至不可能的。生成器和迭代器的用例范围从简单的（例如生成数字序列）到复杂的（例如生成斐波那契数的无限序列）。了解如何使用生成器和迭代器可以帮助您在 Python 中编写更高效和内存友好的代码。