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

可迭代对象和迭代器：

"""
可迭代对象包括迭代器、序列（字符串、列表、元组）和字典
可迭代对象：要么实现__iter__方法，返回该可迭代对象的迭代器类实例，要么实现了__getitem__方法，并且参数是从零开始的索引
迭代器：实现了__iter__方法，返回迭代器本身，且实现了__next__方法，返回下一个元素，若没有元素抛出StopIteration异常
"""


class MyList(object):                       # 定义可迭代对象类(__iter__方法)
    def __init__(self, num):
        self.data = num

    def __iter__(self):
        return MyListIterator(self.data)    # 返回该可迭代对象的迭代器类实例


class MyListIterator(object):              # 定义迭代器类(__iter__方法和__next__方法)
    def __init__(self, data):
        self.data = data
        self.now = 0

    def __iter__(self):
        return self                         # 返回迭代器本身

    def __next__(self):
        while self.now < self.data:
            self.now += 1
            return self.now-1               # 返回下一个元素
        raise StopIteration


def main():
    my_list = MyList(5)
    print(type(my_list))

    my_list_iter = iter(my_list)
    print(type(my_list_iter))

    for i in my_list:
        print(i)


if __name__ == '__main__':
    main()

"""
结果：
<class '__main__.MyList'>
<class '__main__.MyListIterator'>
0
1
2
3
4
"""

生成器：

"""
生成器：一种特殊的迭代器，自动实现了“迭代器协议”（__iter__和__next__方法）
"""


def my_generator(num):                  # 定义生成器
    now = 0
    while now < num:
        temp = (yield now)
        now = now + 1 if temp is None else temp


def main():
    my_gen = my_generator(5)
    print(type(my_gen))
    print(dir(my_gen))                    # 返回该对象拥有的方法名

    print(next(my_gen))
    print(next(my_gen))

    my_gen.send(3)                        # 重置当前迭代值
    print("my_gen.send(3)")
    print(next(my_gen))


if __name__ == '__main__':
    main()

"""
结果：
<class 'generator'>
['__class__', '__del__', '__delattr__', '__dir__', '__doc__', '__eq__', '__format__', '__ge__', '__getattribute__', '__gt__', '__hash__', '__init__', '__init_subclass__', '__iter__', '__le__', '__lt__', '__name__', '__ne__', '__new__', '__next__', '__qualname__', '__reduce__', '__reduce_ex__', '__repr__', '__setattr__', '__sizeof__', '__str__', '__subclasshook__', 'close', 'gi_code', 'gi_frame', 'gi_running', 'gi_yieldfrom', 'send', 'throw']
0
1
my_gen.send(3)
4
"""