如何用Python实现可重用的自定义range类

在Python的标准库中，range类是非常常用的迭代工具，它可以生成指定范围的整数序列，并且支持步长设置，使用起来十分方便。但内置的range类的行为是固定的，当我们需要实现一些特殊的迭代逻辑，比如反向迭代、带条件过滤的迭代时，就需要自己实现一个可重用的自定义range类。

自定义range类的核心原理

要实现可重用的自定义range类，首先需要理解Python的迭代器协议。一个可迭代对象需要实现__iter__方法，该方法返回迭代器对象；而迭代器对象需要实现__iter__和__next__两个方法，其中__next__方法负责返回下一个迭代值，直到迭代结束抛出StopIteration异常。

内置的range类每次迭代都会返回一个新的迭代器，因此可以被多次复用。我们的自定义range类也需要遵循这个逻辑，确保每次调用__iter__方法时都返回一个新的迭代器实例，而不是复用同一个迭代器，这样才能保证类的可重用性。

基础版自定义range类实现

我们先实现一个基础版本的自定义range类，支持起始值、结束值和步长的设置，功能类似内置的range类。

class MyRange:
    def __init__(self, start, end, step=1):
        # 初始化参数，校验步长不能为0
        if step == 0:
            raise ValueError("步长不能为0")
        self.start = start
        self.end = end
        self.step = step

    def __iter__(self):
        # 返回迭代器实例，每次调用都返回新的迭代器，保证可重用
        return MyRangeIterator(self.start, self.end, self.step)


class MyRangeIterator:
    def __init__(self, start, end, step):
        self.current = start
        self.end = end
        self.step = step

    def __iter__(self):
        # 迭代器自身的__iter__返回自身
        return self

    def __next__(self):
        # 判断迭代是否结束，根据步长正负调整判断逻辑
        if self.step > 0 and self.current >= self.end:
            raise StopIteration
        if self.step < 0 and self.current <= self.end:
            raise StopIteration
        # 保存当前值作为返回结果
        result = self.current
        # 更新当前值
        self.current += self.step
        return result

代码逻辑说明

上面的代码分为两个类，MyRange是可迭代对象，负责接收用户传入的参数，并且每次调用__iter__方法时返回一个新的MyRangeIterator迭代器实例。MyRangeIterator是迭代器类，负责具体的迭代逻辑，每次调用__next__方法就返回下一个符合规则的值，直到迭代结束抛出StopIteration异常。

这里需要注意步长的正负判断，当步长为正数时，当前值大于等于结束值就停止迭代；当步长为负数时，当前值小于等于结束值就停止迭代，这样的逻辑和内置range类的行为保持一致。

自定义range类的可重用性验证

我们可以通过多次迭代同一个自定义range实例来验证它的可重用性，代码如下：

# 创建自定义range实例
my_range = MyRange(0, 5, 1)
# 第一次迭代
print("第一次迭代结果：")
for num in my_range:
    print(num)
# 第二次迭代
print("第二次迭代结果：")
for num in my_range:
    print(num)

运行上面的代码，会两次输出0到4的数字，说明同一个MyRange实例可以被多次迭代，实现了可重用的效果。如果使用单迭代器的实现方式，第二次迭代就不会有任何输出，这就是区分可迭代对象和迭代器的重要性。

扩展自定义range类的功能

基础版本的功能和内置range类似，我们可以根据实际需求扩展功能，比如添加反向迭代的支持，或者添加过滤条件。下面我们实现一个支持反向迭代的自定义range类：

class AdvancedMyRange:
    def __init__(self, start, end, step=1):
        if step == 0:
            raise ValueError("步长不能为0")
        self.start = start
        self.end = end
        self.step = step

    def __iter__(self):
        # 正向迭代返回正向迭代器
        return MyRangeIterator(self.start, self.end, self.step)

    def __reversed__(self):
        # 反向迭代返回反向迭代器，步长取反，起始和结束值互换
        return MyRangeIterator(self.end - self.step, self.start - self.step, -self.step)


# 复用之前的MyRangeIterator类，不需要重复定义

测试反向迭代的代码：

adv_range = AdvancedMyRange(0, 5, 1)
print("正向迭代结果：")
for num in adv_range:
    print(num)
print("反向迭代结果：")
for num in reversed(adv_range):
    print(num)

运行后会先输出0到4的正向序列，再输出4到0的反向序列，扩展了自定义range类的使用场景。

注意事项

• 自定义range类时，一定要将可迭代对象和迭代器的逻辑分开，确保__iter__方法每次返回新的迭代器实例，否则类无法被多次复用。
• 处理步长为负数的场景时，迭代结束的判断条件要和步长为正数的场景区分开，避免出现迭代逻辑错误。
• 如果需要在自定义range类中添加更多功能，比如条件过滤，可以在迭代器的__next__方法中添加判断逻辑，跳过不符合条件的值即可。

通过上面的实现方式，我们可以根据实际需求灵活定制自定义range类的行为，并且保证类的可重用性，满足不同场景下的迭代需求。

Python自定义range类可重用迭代器__iter____next__修改时间：2026-06-12 11:30:23