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就很有用。