day9_python迭代器和生成器

迭代器

class MyIterator:
    '''供可迭代对象 变成一个 迭代器'''

    def __init__(self, mylist):
        self.mylist = mylist
        # 当前访问的位置current
        self.current = 0

    # 重写 __next__ 的是迭代器
    def __next__(self):
        current = self.current
        self.current += 1
        if current < len(self.mylist.items):
            item = self.mylist.items[current]
            return item
        else:
            raise StopIteration

    # 记住，就这样写
    def __iter__(self):
        return self

可迭代对象

class MyList:
    '''自定义一个可迭代对象'''

    def __init__(self):
        self.items = []

    def add(self, value):
        self.items.append(value)

    # 重写 __iter__ 方法的就是可迭代对象,要求返回值是一个迭代器
    def __iter__(self):
        myIterator = MyIterator(self)
        return myIterator

完整程序 及 运行结果

from collections.abc import Iterable, Iterator


class MyList:
    '''自定义一个可迭代对象'''

    def __init__(self):
        self.items = []

    def add(self, value):
        self.items.append(value)

    # 重写 __iter__ 方法的就是可迭代对象,要求返回值是一个迭代器
    def __iter__(self):
        myIterator = MyIterator(self)
        return myIterator


class MyIterator:
    '''供可迭代对象 变成一个 迭代器'''

    def __init__(self, mylist):
        self.mylist = mylist
        # 当前访问的位置current
        self.current = 0

    # 重写 __next__ 的是迭代器
    def __next__(self):
        current = self.current
        self.current += 1
        if current < len(self.mylist.items):
            item = self.mylist.items[current]
            return item
        else:
            raise StopIteration

    # 记住，就这样写
    def __iter__(self):
        return self


if __name__ == '__main__':
    # Iterable 可迭代的
    # Iterator 迭代器
    mylist = MyList()  # 创建一个可迭代对象
    # isinstance(x,y)  判断 x是否为 y类型
    # 可迭代对象不一定是 迭代器
    print(isinstance(mylist, Iterable))  # True
    print(isinstance(mylist, Iterator))  # False

    mylist.add(1)
    mylist.add(2)
    mylist.add(3)

    # 可迭代对象 就可以用for遍历
    for i in mylist:
        print(i)

    print(" - " * 50)

    # 迭代器 必然是可迭代对象
    print(isinstance(MyIterator(mylist), Iterator))  # True
    print(isinstance(MyIterator(mylist), Iterable))  # True
    # 迭代器也能用 for 遍历
    for i in MyIterator(mylist):
        print(i)

生成器-generator（支持next）

yield关键字

# 含有 yield 关键字的方法叫生成器
def fib(n):
    current = 0
    num1 = num2 = 1
    while current < n:
        num = num1
        num1, num2 = num2, num1 + num2
        current += 1
        yield num
    return 'done'

if __name__ == '__main__':
    F = fib(10)
    print(F)
    print("*" * 50)
    for i in range(10):
        print(next(F), end=' ')

send()函数 唤醒生成器的同时，向断点处传入一个附加数据

def gen():
    i = 0
    while i < 5:
        temp = yield i
        print(temp)
        i += 1


if __name__ == '__main__':
    g = gen()
    # 程序在yield 卡住，执行next 会返回当前值，但是需要打印
    # None 是temp打印的   数字是print 打印的
    print(next(g))  # 0
    next(g)  # None,因为没传参数
    print(next(g))  # None 2
    print(g.send("hello"))  # hello 3