在Python中如何自定义类似range的迭代器

avatar
作者
筋斗云
阅读量:0

在Python中,要创建一个类似range的迭代器,你需要定义一个类,并实现__iter__()__next__()方法。以下是一个简单的示例:

class CustomRange:     def __init__(self, start, end, step=1):         self.start = start         self.end = end         self.step = step      def __iter__(self):         return self      def __next__(self):         if self.start >= self.end:             raise StopIteration         current = self.start         self.start += self.step         return current  # 使用示例 custom_range = CustomRange(0, 10, 2) for i in custom_range:     print(i) 

这个示例中,我们创建了一个名为CustomRange的类,它接受startendstep作为参数。__iter__()方法返回当前对象(即迭代器对象),而__next__()方法在每次调用时返回序列中的下一个值,直到达到end值。当没有更多元素可供迭代时,__next__()方法将引发StopIteration异常,表示迭代完成。

广告一刻

为您即时展示最新活动产品广告消息,让您随时掌握产品活动新动态!