python3 迭代器与生成器

pythom3 迭代器与生成器

迭代器
'''
迭代器是python最强大的功能之一,是访问集合元素的一种方式。
迭代器是一个可以记住遍历的位置对象
迭代器对象从集合的第一个元素开始访问,直到所有的元素被访问玩结束。迭代器只能往前不会后退。
迭代器有两个基本方法:iter()和next().
字符串,列表或元组对象都可用于创建迭代器:

'''
list=[1,2,6,3]
it=iter(list)#创建迭代器对象
print(next(it))#结果1
print(next(it))#结果2
print(next(it))#结果6
#根据上面的,就可以推断打印一次就可以输出迭代器的下一个元素

for i in it:#利用for循环依次迭代输出
    print(i)
#也可以使用next()函数
import sys #引入sys模块
li=[1,2,365,9]
ip=iter(li)#创建迭代器对象
while True:
    try:
        print(next(ip))
    except StopIteration:
        sys.exit()
#结果1,2,365,9

生成器yield
'''
在python中,使用了yield的函数被称为生成器(generator)
跟普通函数不同的是,生成器是一个返回迭代器的函数,只能用于迭代操作,更简单点理解生成器就是一个迭代器。
在调用生成器的运行过程中,每次遇到yield时函数会暂停并保存当前所有运行信息,返回yield的值。
并在下一次执行next()方法时从当前位置继续运行
'''
#使用yield实现斐波那契数列
import sys
def fibonacci(n):#生成函数 斐波那契
    a,b,counter=0,1,0
    print('a',a)
    while True:
        if(counter>n):
            return
        yield a
        a,b=b,a+b
        counter+=1
f=fibonacci(10)#f是一个迭代器由生成器生成
while True:
    try:
        print(next(f),end=' ')
    except StopIteration:
        sys.exit()
#结果是0,1,1,2,3,5,8,13,21,34,55

总结;

什么情况下需要使用yield

    一个函数f,f返回一个list,这个list是动态计算出来的(不管是数学上的计算还是逻辑上的读取格式化),

    并且这个list会很大(无论是固定很大还是随着输入参数的增大而增大),这个时候,我们希望每次调用这个函数并使用迭代器进行

    循环的时候一个一个的到每个list元素而不是直接得到一个完整的list来节省内存,这个时候yield就很有用。

原文地址:https://www.cnblogs.com/QI1125/p/7564988.html