python高级——迭代器、生成器

朱雀 2022-01-07 11:43 329阅读 0赞

# 1.迭代器 #

迭代是访问集合元素的一种方式。迭代器是一个可以记住遍历的位置的对象。迭代器对象从集合的第一个元素开始访问，直到所有

# 2.iter()函数与next()函数 #

list、tuple等都是可迭代对象，我们可以通过iter()函数获取这些可迭代对象的迭代器。然后我们可以对获取到的迭代器不断使用next()函数来获取下一条数据。iter()函数实际上就是调用了可迭代对象的\_\_iter\_\_方法。

# 3.斐波那契数列迭代器 #

class FibIterator(object):
        """斐波那契数列迭代器"""
        def __init__(self, n):
            """ :param n: int, 指明生成数列的前n个数 """
            self.n = n
            # current用来保存当前生成到数列中的第几个数了
            self.current = 0
            # num1用来保存前前一个数，初始值为数列中的第一个数0
            self.num1 = 0
            # num2用来保存前一个数，初始值为数列中的第二个数1
            self.num2 = 1
    
        def __next__(self):
            """被next()函数调用来获取下一个数"""
            if self.current < self.n:
                num = self.num1
                self.num1, self.num2 = self.num2, self.num1+self.num2
                self.current += 1
                return num
            else:
                raise StopIteration
    
        def __iter__(self):
            """迭代器的__iter__返回自身即可"""
            return self
    
    
    if __name__ == '__main__':
        fib = FibIterator(10)
        for num in fib:
            print(num, end=" ")

# 4.生成器 #

使用了yield关键字的函数不再是函数，而是生成器。（使用了yield的函数就是生成器）  
yield关键字有两点作用：  
保存当前运行状态（断点），然后暂停执行，即将生成器（函数）挂起  
将yield关键字后面表达式的值作为返回值返回，此时可以理解为起到了return的作用  
可以使用next()函数让生成器从断点处继续执行，即唤醒生成器（函数）  
Python3中的生成器可以使用return返回最终运行的返回值，而Python2中的生成器不允许使用return返回一个返回值（即可以使用return从生成器中退出，但return后不能有任何表达式）。