迭代器与生成器
迭代器
迭代是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
生成器
在 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