Python 迭代器

迭代器是一种对象,该对象包含值的可计数数字。

迭代器是可迭代的对象,这意味着您可以遍历所有值。

从技术上讲,在 Python 中,迭代器是实现迭代器协议的对象,它包含方法 __iter__() 和 __next__()。

 

迭代器 VS 可迭代对象(Iterable)

列表、元组、字典和集合都是可迭代的对象。它们是可迭代的容器,您可以从中获取迭代器(Iterator)。

所有这些对象都有用于获取迭代器的 iter() 方法:

mytuple=('apple','banana','cherry')


#iter() 函数用来生成迭代器。法
#next() 返回迭代器的下一个项目。
#next() 函数要和生成迭代器的 iter() 函数一起使用。
myit=iter(mytuple)
print(next(myit))#apple
print(next(myit))#banana
print(next(myit))#cherry

#遍历迭代器
for x in mytuple:
    print(x)

#for 循环实际上创建了一个迭代器对象,并为每个循环执行 next() 方法。




#创建迭代器

#要把对象/类创建为迭代器,必须为对象实现 __iter__() 和 __next__() 方法。

#正如您在 Python 类/对象 一章中学到的,所有类都有名为 __init__() 的函数,它允许您在创建对象时进行一些初始化。

#__iter__() 方法的作用相似,您可以执行操作(初始化等),但必须始终返回迭代器对象本身。

#__next__() 方法也允许您执行操作,并且必须返回序列中的下一个项目。

#创建一个返回数字的迭代器,从 1 开始,每个序列将增加 1(返回 1、2、3、4、5 等):

class MyNumbers:
    def __iter__(self):
        self.a=1
        return self
    
    def __next__(self):
        x=self.a
        self.a+=1
        return x

myclass=MyNumbers()
myiter=iter(myclass)


print(next(myiter))#1
print(next(myiter))#2
print(next(myiter))#3
print(next(myiter))#4
print(next(myiter))#5


#StopIteration
#如果你有足够的 next() 语句,或者在 for 循环中使用,则上面的例子将永远进行下去。
#为了防止迭代永远进行,我们可以使用 StopIteration 语句。
#在 __next__() 方法中,如果迭代完成指定的次数,我们可以添加一个终止条件来引发错误:
class YouNumbers:
    def __iter__(self):
        self.a=1
        return self
    
    def __next__(self):
        if self.a<=20:
            x=self.a
            self.a+=1
            return x
        else:
            raise StopIteration

youclass=YouNumbers()
youiter=iter(youclass)

for x in youiter:
    print(x)
View Code

 

posted @ 2021-08-12 14:46  还有什么值得拥有  阅读(30)  评论(0编辑  收藏  举报