迭代器
迭代是Python最强大的功能之一,是访问集合元素的一种方式
可以被
next()
函数调用并不断返回下一个值的对象称为迭代器:Iterator迭代器是一个可以记住遍历的位置的对象
迭代器对象从集合的第一个元素开始访问,直到所有的元素被访问完结束。迭代器只能往前不会后退
迭代器有两个基本的方法:
iter()
和next()
集合数据类型,如
list
、tuple
、dict
、set
、str
等对象都可用于创建迭代器迭代器对象可以使用常规for语句进行遍历
代码示例:
用
iter()
创建迭代器
list=[1,2,3,4]
it = iter(list) # 创建迭代器对象
for x in it:
print (x, end=" ") # 输出结果 1 2 3 4
输出结果如下:
1 2 3 4
使用
next()
函数获取元素
import sys # 引入 sys 模块, 此处用于结束循环函数
list = [1, 2, 3, 4]
it = iter(list) # 创建迭代器对象
while True:
try:
print(next(it))
except StopIteration:
sys.exit() # 不打印时结束循环
输出结果如下:
1
2
3
4
生成器
在 Python 中,使用了
yield
的函数被称为生成器(generator)跟普通函数不同的是,生成器是一个返回迭代器的函数,只能用于迭代操作,更简单点理解生成器就是一个迭代器
在调用生成器运行的过程中,每次遇到
yield
时函数会暂停并保存当前所有的运行信息,返回yield
的值。并在下一次执行next()
方法时从当前位置继续运行generator 保存的是算法,每次调用
next(n)
,就计算出n
的下一个元素的值,直到计算到最后一个元素
代码示例:
使用 yield 实现斐波那契数列
import sys
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(10) # f 是一个迭代器,由生成器返回生成
while True:
try:
print (next(f), end=" ")
except StopIteration:
sys.exit()
输出结果如下:
0 1 1 2 3 5 8 13 21 34 55