在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
的类,它接受start
、end
和step
作为参数。__iter__()
方法返回当前对象(即迭代器对象),而__next__()
方法在每次调用时返回序列中的下一个值,直到达到end
值。当没有更多元素可供迭代时,__next__()
方法将引发StopIteration
异常,表示迭代完成。