目录
迭代器
迭代器的创建与使用
生成器(generator)
yield
推导式变生成器
迭代器在迭代元素的时候可以记住遍历的位置,迭代器只能前进不能后退。
创建迭代器:iter() 括号中的参数为可迭代对象
迭代器结果输出:next() 括号中的参数为创建出来的迭代器
迭代器是一个类,迭代器类中定义有__iter__() 和__next__()两个方法。__iter__()方法用来返回一个迭代器对象,这个迭代器对象实现了__next__()方法并通过 StopIteration 异常标识迭代的完成。__next__() 方法会返回下一个迭代器对象。我们可以自己创建迭代器类来使用。
使用迭代器时,如果迭代器中数据迭代完之后继续迭代就会报错。因此在使用next()迭代时,一定要做异常的处理。
import sys # 引入sys模块 while True: try: print(next(迭代器)) except StopIteration: sys.exit()使用了yield的函数被称为生成器。生成器是一个可以返回迭代器的函数,这个生成器函数只能用来做迭代操作,所以生成器可以看作是迭代器。生成器中yield的作用是做暂停操作,并返回一个yield值(这个值用来存储当前的运行信息),在下一次next()时从返回的yield处继续运行。yield支持显示的暂停和恢复,也支持隐式的迭代协议,yield可用在多线程中。
示例:创建一个斐波那契数列生成器
import sys def fibonacci(): def fi(n,x,y): for _ in range(n): yield x x,y = y,x+y while True: try: x = int(input('请输入第一个数:')) y = int(input('请输入第二个数:')) n = int(input('请输入生成的数列总数:')) f = fi(n, x, y) while True: try: print(next(f), end=' ') except StopIteration: sys.exit() except ValueError: print('只能输入数字,请重新输入:') fibonacci()将外层的括号写成小括号即为一个生成器
list_generator = (x for x in range(1,21) if x%2==0) for _ in range(5): # 使用循环迭代五次 print(next(list_generator),end=' ')