python：迭代器详解及使用方法

目录

 

一、什么是迭代器

二、创建迭代器

三、迭代器方法

四、迭代器的应用：通过迭代器读取视频帧

---

一、什么是迭代器

迭代，字面理解就是多次重复做一件事情，类似于循环。迭代器是实现了__next__()方法的对象，并且这个方法在调用时不需要任何参数，它是访问可迭代序列的一种方式，通常从序列的第一个元素开始访问，直到所有的元素被访问才结束。

迭代器只能前进不能后退。

使用迭代器不需要事先准备好整个迭代过程中的所有元素。迭代器仅仅在迭代到某个元素时才计算该元素，儿在这之前或之后可以不存在或者被销毁。因此迭代器适合遍历一些数量巨大甚至无限的序列。

二、创建迭代器

• 使用内置函数iter(iterable)可以将可迭代序列转换为迭代器

alist = [2, 5, 8, 11]
arr = (2, 5, 8, 11)
s = 'china'

print(iter(alist))
print(iter(arr))
print(iter(s))

# 输出
# <list_iterator object at 0x10d9e9240>
# <tuple_iterator object at 0x10d9e9240>
# <str_iterator object at 0x10d9e9240>

• 自定义迭代器

python中的迭代器本质上每次调用__next__()方法都返回下一个元素或抛出StopIteration的容器对象

由于python中没有“迭代器”这个类，因袭具有以下两个特性的类都可以称为“迭代器”类

1. 有__next__()方法，返回容器的下一个元素或抛出StopIteration异常
2. 有__iter__()方法，返回迭代器本身

#斐波那契数列
class Fabs():
    def __init__(self,max):
        self.max=max
        self.n,self.a,self.b=0,0,1

    def __iter__(self):#定义__iter__方法
        return self

    def __next__(self):#定义__next__方法
        if self.n<self.max:
            tmp=self.b
            self.a,self.b=self.b,self.a+self.b
            #等价于：
            #t=(self.a,self.a+self.b)
            #self.a=t[0]
            #self.b=t[1]
            self.n+=1
            return tmp
        raise StopIteration

print(Fabs(5))
for item in Fabs(10):
    print(item,end=' ')




# 输出
# <__main__.Fabs object at 0x1034d2278>
# 1 1 2 3 5 8 13 21 34 55 

三、迭代器方法

iter.__next__()：返回迭代器的下一个元素，但没有下一个元素时抛出StopIteration异常

list=[1,2,3,4]
list=iter(list)
print(list.__next__())
print(list.__next__())
print(list.__next__())
print(list.__next__())
print(list.__next__())



# 输出
# 1
# 2
# 3
# 4
# Traceback (most recent call last):
#   File "/Users/dwz/Downloads/learning2.py", line 702, in <module>
#     print(list.__next__())
# StopIteration

iter.__iter__():返回迭代器对象本身

list=[1,2,3,4]
list=iter(list)
print(list.__iter__())


# 输出
# <list_iterator object at 0x10cdd8240>

四、迭代器的应用：通过迭代器读取视频帧

class VideoIter():
    def __init__(self, video_name):
        self.cap = cv2.VideoCapture(video_name)
        self.num_frame = self.cap.get(cv2.CAP_PROP_FRAME_COUNT)
        # self.iter = iter(self.cap)
        self.index = -1

    def __iter__(self):
        return self

    def __next__(self):
        # while True:
        if self.index < self.num_frame:
            ret, frame = self.cap.read()
            if ret:
                self.index += 1
                return frame
            else:
                raise ValueError
        else:
            raise StopIteration


index = 100   # 要查找的帧
video_iter = VideoIter(video_path)
while video_iter.index != index and video_iter.index < num_frame:
      frame = next(video_iter)