首页 > > 网络编程 > 其它 >

迭代器执行流程

2019-02-17 01:54:15来源：博客园阅读 ()

迭代器的执行流程，以及说明可迭代对象不一定是迭代器，但迭代器一定是可迭代对象

实例1

from collections import Iterable, Iterator
import time


class Classmate(object):
    """可迭代的对象（必须存在__iter__）"""
    def __init__(self):
        self.names = list()

    def add(self, name):
        self.names.append(name)

    def __iter__(self):
        """
        如果想要一个对象称为一个可以迭代的对象，即可以使用for，那么必须实现__iter__方法
        如return引用的是一个迭代器，则返回迭代器里__next__的返回值
        """
        return ClassIterator(self)


class ClassIterator(object):
    """迭代器（必须同时存在__iter__和__next__）"""

    def __init__(self, obj):
        self.obj = obj
        self.current_num = 0

    def __iter__(self):
        pass

    def __next__(self):
        """for循环取出的name来自__next__的返回值"""

        if self.current_num < len(self.obj.names):
            ret = self.obj.names[self.current_num]
            self.current_num += 1
            return ret
        else:
            # 越界抛出异常退出
            raise StopIteration


classmate = Classmate()

classmate.add("熊")
classmate.add("猫")
classmate.add("狗")

# print("判断classmate是否是可以迭代的对象：", isinstance(classmate, Iterable))
# classmate_tierator = iter(classmate)
# print("判断classmate_tierator是否是迭代器：", isinstance(classmate_tierator, Iterator))
# print(next(classmate_tierator))

# 1.for循环会先判断 classmate是否是可迭代的对象，若成立执行下面步骤
# 2.for循环调用对象中的__iter__获取其返回值，如返回值是可迭代对象，执行下面步骤
# 3.for每循环一次，则会调用迭代器里的__next__，获取其返回值，如未设置退出条件，则不停返回None
for name in classmate:
    print(name)
    time.sleep(1)

实例1的优化

from collections import Iterable, Iterator
import time


class Classmate(object):
    """
    迭代器
    存在__iter__为可迭代对象
    同时存在__iter__和__next__为迭代器
    """
    def __init__(self):
        self.names = list()
        self.current_num = 0

    def add(self, name):
        self.names.append(name)

    def __iter__(self):
        """
        如果想要一个对象称为一个可以迭代的对象，即可以使用for，那么必须实现__iter__方法
        如return返回的是一个迭代器，则返回迭代器里__next__的返回值
        """
        return self

    def __next__(self):
        """for循环取出的name来自__next__的返回值"""

        if self.current_num < len(self.names):
            ret = self.names[self.current_num]
            self.current_num += 1
            return ret
        else:
            # 越界抛出异常退出
            raise StopIteration


classmate = Classmate()

classmate.add("熊")
classmate.add("猫")
classmate.add("狗")

# 1.for循环会先判断 classmate是否是可迭代的对象，若成立执行下面步骤
# 2.for循环调用对象中的__iter__获取其返回值，如返回值是可迭代对象，执行下面步骤
# 3.for每循环一次，则会调用迭代器里的__next__，获取其返回值，如未设置退出条件，则不停返回None
for name in classmate:
    print(name)
    time.sleep(1)