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

   2024-09-30 8900
核心提示:在Python中,要创建一个类似range的迭代器,你需要定义一个类,并实现__iter__()和__next__()方法。以下是一个简单的示例:class

在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异常,表示迭代完成。

 
举报打赏
 
更多>同类物流大全
推荐图文
推荐物流大全
点击排行

网站首页  |  关于我们  |  联系方式网站留言    |  赣ICP备2021007278号