迭代器的概念

最新推荐文章于 2022-09-14 15:49:32 发布

原创最新推荐文章于 2022-09-14 15:49:32 发布 · 290 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #tensorflow #开发语言

python概念理解专栏收录该内容

5 篇文章

订阅专栏

本文探讨了迭代器在编程中的作用，重点讲解了如何创建迭代器类以返回1到n的平方，以及迭代器如何通过计算节省空间，同时提及了这种优化可能带来的性能影响。通过实例演示了myiter类的__iter__和__next__方法的使用及其StopIteration异常处理。

迭代器必须满足两个条件，一个是可迭代，一个是可以让next()调用，例如列表，元组，字符串等都是可以迭代的，但无法让next()调用，因此不属于迭代器，迭代器的主要好处在于节省空间。

例如[1,2,3,4]可以调任意值，原因是内存中开辟了四块空间，如果是迭代器的话，一般是从头元素开始，如果得到下一个元素，通过某种计算可以得到下个元素，第一个元素则消失不见，所以总的来说只开辟了一块空间，从而节省了空间，但会影响性能。

list=[1,2,3,4]
it = iter(list) # 可以将可迭代的对象转化成迭代器，但一般不会这么做，因为list已经占了一部空间，再使用迭代方法也会占用空间，所以这并没有节省空间
print(next(it))
print(next(it))
print(next(it))
print(next(it))
print(next(it))

返回结果：
print(next(it))StopIteration
1
2
3
4
当元素迭代结束后，超出会报一个StopIteration异常

创建一个迭代器

# 迭代器必须包含__iter__，__next__， 主要的计算在next下面
# 案例：输入n，返回1-n的平方，
class myiter():
    def __init__(self,n):
        self.n = n
        self.first = 1

    def __iter__(self):
        return self  # 固定语法

    def __next__(self):
        if self.first>self.n:
            raise StopIteration  # 当迭代器遇到该异常时，则结束
        data = self.first**2
        self.first+=1
        return data

it = myiter(2)
for i in it:
    print(i)

返回结果：
1
4