Python教程：实现可重用自定义range类

本文深入探讨了python中自定义迭代器与内置`range`行为差异的根源。`range`对象的可重用性源于其类实现，每次迭代请求都会生成新的迭代器，而简单的生成器函数则创建单次使用的迭代器。教程将指导您通过构建一个自定义类并实现`__iter__`方法，来创建与内置`range`功能相似的可重用自定义范围对象。

理解Python中的迭代器与可迭代对象

在Python中，迭代是一个核心概念。它允许我们遍历序列（如列表、元组、字符串）或其他集合。要理解如何创建可重用的自定义范围，首先需要区分两个关键概念：可迭代对象（Iterable）和迭代器（Iterator）。

• 可迭代对象：是任何可以通过for...in循环遍历的对象。它必须实现__iter__方法，该方法返回一个迭代器。常见的可迭代对象包括列表、元组、字符串、字典、集合以及内置的range对象。
• 迭代器：是表示数据流的对象。它必须实现__iter__方法（返回自身）和__next__方法。__next__方法在每次调用时返回序列中的下一个元素，当没有更多元素时，它会抛出StopIteration异常。

生成器函数（使用yield关键字的函数）在被调用时会返回一个生成器迭代器对象。这个生成器迭代器本身就是一种迭代器。

生成器函数的局限性：单次使用

考虑以下使用生成器函数实现的自定义范围示例：

def exampleCustomRange(stopExclusive):
  """
  一个简单的生成器函数，模拟range的行为。
  """
  for i in range(stopExclusive):
    yield i

# 实例化生成器迭代器
customRange = exampleCustomRange(3)

# 第一次迭代：正常工作
print([x for x in customRange]) # 输出: [0, 1, 2]

# 第二次迭代：结果为空
print([x for x in customRange]) # 输出: []

正如示例所示，customRange在第一次迭代后就被“耗尽”了。这是因为exampleCustomRange(3)调用返回的是一个生成器迭代器。这个迭代器在第一次遍历时，其内部状态会记录当前位置，并在每次yield后暂停。当所有值都被yield出来后，迭代器就达到了终点，再次尝试迭代时，它已经没有更多值可以生成了，因此会立即抛出StopIteration，导致列表推导式返回一个空列表。

立即学习“Python免费学习笔记（深入）”；

这与内置的range对象的行为形成了鲜明对比：

builtinRange = range(3)

# 第一次迭代：正常工作
print([x for x in builtinRange]) # 输出: [0, 1, 2]

# 第二次迭代：依然正常工作，可重用
print([x for x in builtinRange]) # 输出: [0, 1, 2]

内置的range对象能够重复使用，是因为它是一个可迭代对象，而不是一个直接的迭代器。每次对其进行迭代时（例如，在for循环中或使用列表推导式），它都会返回一个新的迭代器。

MarsX

AI驱动快速构建App，低代码无代码开发，改变软件开发的游戏规则

159

查看详情

解决方案：实现一个可重用的自定义类

要实现与range类似的可重用行为，我们需要创建一个类，使其成为一个可迭代对象。这个类的关键在于实现__iter__方法。当对该类的实例进行迭代时，Python会调用其__iter__方法，该方法应该返回一个新的迭代器。

以下是实现可重用自定义范围的正确方法：

class ExampleCustomRange:
    """
    一个可重用的自定义范围类，模拟内置range的行为。
    """
    def __init__(self, stop_exclusive):
        """
        初始化自定义范围的上限（不包含）。
        """
        self.stop_exclusive = stop_exclusive

    def __iter__(self):
        """
        每次迭代请求时，此方法都会被调用，并返回一个新的迭代器。
        这里我们使用一个生成器来作为迭代器。
        """
        for i in range(self.stop_exclusive):
           yield i

# 实例化自定义范围类
reusable_custom_range = ExampleCustomRange(3)

# 第一次迭代：正常工作
print("第一次迭代:", [x for x in reusable_custom_range]) # 输出: 第一次迭代: [0, 1, 2]

# 第二次迭代：依然正常工作，可重用
print("第二次迭代:", [x for x in reusable_custom_range]) # 输出: 第二次迭代: [0, 1, 2]

# 也可以在for循环中使用
print("通过for循环迭代:")
for num in reusable_custom_range:
    print(num, end=" ") # 输出: 0 1 2
print("\n再次通过for循环迭代:")
for num in reusable_custom_range:
    print(num, end=" ") # 输出: 0 1 2
print()

在这个ExampleCustomRange类中：

1. __init__(self, stop_exclusive) 方法用于存储范围的上限（不包含）。这是可迭代对象的状态。
2. __iter__(self) 方法是核心。每次当Python需要一个迭代器来遍历ExampleCustomRange的实例时（例如，当您将其放入for循环或列表推导式中），就会调用此方法。
3. __iter__ 方法内部是一个生成器表达式for i in range(self.stop_exclusive): yield i。每次调用__iter__时，它都会创建一个新的生成器迭代器并返回。这个新的生成器迭代器拥有自己独立的迭代状态，因此可以从头开始生成序列。

通过这种方式，reusable_custom_range对象本身并没有被“耗尽”，因为它每次被迭代时都会提供一个新的、未耗尽的迭代器。

总结与注意事项

• 可迭代对象 vs. 迭代器：理解这两者的区别是关键。可迭代对象是能够产生迭代器的对象（通过__iter__方法），而迭代器是实际执行迭代并保持状态的对象（通过__next__方法）。
• 生成器函数的用途：生成器函数在需要一次性、按需生成序列时非常有用，例如处理大型文件或无限序列，以节省内存。它们返回的是迭代器。
• 类的用途：当需要一个可重用、可多次迭代的对象时，应该定义一个类，并实现其__iter__方法来返回一个新的迭代器（通常是一个生成器）。
• 设计原则：一个好的可迭代对象应该在每次迭代请求时提供一个“新鲜”的迭代器，这样它就可以被多次遍历。

通过遵循上述原则，您可以创建出行为与内置range对象相似的自定义可重用范围类，从而更好地控制Python中的迭代逻辑。

以上就是Python教程：实现可重用自定义range类的详细内容，更多请关注php中文网其它相关文章！