可迭代对象、迭代器、生成器

from collections import Iterator, Iterable


class MyListIterator(object):  # 定义迭代器类,其是MyList可迭代对象的迭代器类

    def __init__(self, data):
        self.data = data  # 上边界
        self.now = 0  # 当前迭代值,初始为0

    def __iter__(self):
        return self  # 返回该对象的迭代器类的实例;因为自己就是迭代器,所以返回self

    def __next__(self):  # 迭代器类必须实现的方法
        while self.now < self.data:
            self.now += 1
            return self.now - 1  # 返回当前迭代值  
        raise StopIteration  # 超出上边界,抛出异常


my_list = MyListIterator(5)  # 得到一个可迭代对象
print(type(my_list))  # 返回该对象的类型 <class '__main__.MyListIterator'>
my_list_iter = iter(my_list)  # 得到该对象的迭代器实例,iter函数在下面会详细解释 <class '__main__.MyListIterator'>
print(type(my_list_iter))  # <class '__main__.MyListIterator'>
for i in my_list:  # 一次调用__next__方法
    print(i) # 0,1,2,34
print(isinstance(my_list, Iterator))  # True
print(isinstance(my_list, Iterable))  # True
print(isinstance([], Iterable))  # True
print(isinstance([], Iterator))  # True
gene = (i for i in range(10))  # 创建一个生成器
print(gene, type(gene))  # <generator object <genexpr> at 0x0000024609D9B5C8> <class 'generator'>
print(isinstance(gene, Iterable))  # True
print(isinstance(gene, Iterator))  # True

总结:

  1. 如果一个对象拥有__iter__方法,其是可迭代对象
  2. 定义可迭代对象,必须实现__iter__方法;定义迭代器,必须实现__iter__和next方法
  3. 生成器是一种特殊的迭代器,生成器自动实现了“迭代器协议”(即__iter__和next方法),不需要再手动实现两方法(表现形式为函数里面使用了yield关键字)
posted @ 2022-08-18 22:52  我是小菜鸡丫丫  阅读(20)  评论(0编辑  收藏  举报