本文深入探讨了Python中自定义迭代器与内置`range`行为差异。通过对比生成器函数和类实现，揭示了为何生成器函数创建的迭代器是单次使用的，而`range`对象可重复迭代。核心在于实现一个带有`__iter__`方法的类，使其每次迭代时都能生成一个新的迭代器，从而达到与内置`range`相同的可重用性。
在Python中，迭代是一个核心概念，它允许我们遍历序列、集合或其他可迭代对象中的元素。内置的range函数是生成数字序列的强大工具，其一个显著特性是其可重用性——你可以多次迭代同一个range对象，每次都能得到完整的序列。然而，当尝试使用简单的生成器函数来实现类似自定义范围时，我们常常会遇到一个问题：生成的迭代器只能使用一次。
生成器函数的单次迭代特性
让我们首先通过一个示例来理解这个问题。考虑以下使用生成器函数实现的自定义范围：
def exampleCustomRange(stopExclusive):
for i in range(stopExclusive):
yield i
# 内置 range 的行为
builtinRange = range(3)
print([x for x in builtinRange]) # 输出: [0, 1, 2]
print([x for x in builtinRange]) # 输出: [0, 1, 2] - 可重复使用
# 自定义生成器函数的行为
customRange = exampleCustomRange(3)
print([x for x in customRange]) # 输出: [0, 1, 2]
print([x for x in customRange]) # 输出: [] - 无法重复使用从上述代码中可以看出，builtinRange对象可以被多次迭代并每次都产生相同的结果。而customRange，一个由exampleCustomRange生成器函数创建的迭代器，在第一次迭代耗尽后，第二次迭代时却为空。这是因为exampleCustomRange函数每次调用时返回的是一个生成器对象，这个生成器对象本身就是一个迭代器。迭代器一旦被耗尽，它就无法“重置”或再次生成元素。
立即学习“Python免费学习笔记（深入）”；
实现可重用迭代器的关键：__iter__方法
内置的range对象之所以能够重复迭代，是因为它实际上是一个类的实例，并且遵循了Python的迭代协议。一个可迭代对象（Iterable）是指实现了__iter__方法的对象，该方法必须返回一个新的迭代器。而迭代器（Iterator）是指实现了__iter__和__next__方法的对象，其中__next__方法用于返回序列中的下一个元素。
为了让我们的自定义范围类也具备可重用性，我们需要将其实现为一个类，并在其中定义__iter__方法。这个__iter__方法每次被调用时，都应该返回一个新的迭代器。最简单的方法是让__iter__方法内部使用yield关键字，使其成为一个生成器，从而每次都产生一个新的生成器对象（即新的迭代器）。
构建可重用的自定义范围类
以下是实现可重用自定义范围的正确方法：
class ExampleCustomRange:
"""
一个可重用的自定义范围类，行为类似于内置的 range。
"""
def __init__(self, stop_exclusive):
"""
初始化自定义范围对象。
:param stop_exclusive: 范围的独占上限。
"""
self.stop_exclusive = stop_exclusive
def __iter__(self):
"""
返回一个新的迭代器。
每次对 ExampleCustomRange 实例进行迭代时，都会调用此方法。
"""
print(f"DEBUG: __iter__ called for stop_exclusive={self.stop_exclusive}")
for i in range(self.stop_exclusive):
yield i
# 演示 ExampleCustomRange 的可重用性
print("\n--- 演示 ExampleCustomRange 的可重用性 ---")
reusableCustomRange = ExampleCustomRange(3)
print("第一次迭代:")
print([x for x in reusableCustomRange]) # 输出: [0, 1, 2]
print("第二次迭代:")
print([x for x in reusableCustomRange]) # 输出: [0, 1, 2] - 现在可以重复使用了！
# 进一步测试
print("在 for 循环中迭代:")
for num in reusableCustomRange:
print(num, end=" ") # 输出: 0 1 2
print()
print("创建另一个实例并迭代:")
anotherRange = ExampleCustomRange(5)
print([x for x in anotherRange]) # 输出: [0, 1, 2, 3, 4]在上述ExampleCustomRange类中：
- __init__(self, stop_exclusive): 构造函数用于存储自定义范围的参数，这里是stop_exclusive。这些参数是类的内部状态。
- __iter__(self): 这是核心。当Python试图迭代一个ExampleCustomRange的实例（例如，在一个for循环中，或者使用list()、tuple()、列表推导式等）时，它会调用这个实例的__iter__方法。我们的__iter__方法内部包含了一个for循环和yield语句，这使其成为一个生成器函数。每次调用__iter__时，它都会返回一个全新的生成器对象（即一个新的迭代器），这个新的迭代器会从头开始生成序列。
因此，每次需要迭代reusableCustomRange对象时，Python都会调用reusableCustomRange.__iter__()，并获得一个全新的、尚未耗尽的迭代器。这正是range对象以及其他内置可迭代对象（如列表、元组）的工作方式。
总结
实现可重用的自定义Python range类或任何可迭代对象，关键在于理解迭代器和可迭代对象的区别，并正确实现迭代协议。
- 生成器函数（def func(): yield ...）直接返回一个迭代器。这个迭代器是单次使用的。
- 类如果想成为可迭代对象并支持多次迭代，必须实现__iter__方法。__iter__方法应该返回一个新的迭代器。通常，最简洁的方式是在__iter__方法中使用yield关键字，使其自身成为一个生成器函数，从而每次调用都返回一个新的生成器对象（迭代器）。
通过这种方式，我们不仅能够模拟内置range的可重用行为，还能更深入地理解Python的迭代机制，为构建更复杂、更灵活的数据结构奠定基础。
