1 迭代器
迭代是访问集合元素的一种方式。迭代器是一个可以记住遍历的位置的对象。迭代器对象从集合的第一个元素开始访问,直到所有的元素被访问完结束。迭代器只能往前不会后退。
迭代器有两个基本的方法:iter() 和 next()。
字符串,列表或元组对象都可用于创建迭代器。示例代码如下:
>>> l=[1,2,3,4]
>>> it=iter(l)
>>> print(next(it))
1
>>> print(next(it))
2
>>> print(next(it))
3
>>> print(next(it))
4
迭代器对象可以使用常规for语句进行遍历:
l=[1,2,3,4]
it=iter(l)
for x in it:
print(x,end=' ')
输出如下:
1 2 3 4
使用next()函数进行遍历:
import sys
l=[1,2,3,4]
it=iter(l)
for i in range(len(l)):
print(next(it))
输出如下:
1
2
3
4
2 生成器
在 Python 中,使用了 yield 的函数被称为生成器(generator)。
跟普通函数不同的是,生成器是一个返回迭代器的函数,只能用于迭代操作,更简单点理解生成器就是一个迭代器。
在调用生成器运行的过程中,每次遇到 yield 时函数会暂停并保存当前所有的运行信息,返回 yield 的值, 并在下一次执行 next() 方法时从当前位置继续运行。
调用一个生成器函数,返回的是一个迭代器对象。以下实例使用 yield 实现斐波那契数列:
def fibonacci(n):
a,b,counter=0,1,0
while True:
if(counter>n):
return
yield a
a,b=b,a+b
counter+=1
f=fibonacci(5)
print(list(f))
输出如下:
[0, 1, 1, 2, 3, 5]
参考文献:
Python3 迭代器与生成器