Day 12 :迭代器与生成器
可迭代:在Python中如果一个对象有__iter__( )方法,则称这个对象是可迭代的(Iterable);
其中__iter__( )方法的作用是让对象可以用for ... in循环遍历,列表List、元组Tuple、字典Dictionary、字符串String等数据类型都是可迭代的
迭代器:在Python中如果一个对象有__iter__( )方法和__next__( )方法,则称这个对象是迭代器(Iterator);其中__iter__( )方法是让对象可以用for ... in循环遍历,__next__( )方法是让对象可以通过next(实例名)访问下一个元素。注意:这两个方法必须同时具备,才能称之为迭代器。列表List、元组Tuple、字典Dictionary、字符串String等数据类型虽然是可迭代的,但都不是迭代器,因为他们都没有next( )方法。
1 a = [1,2,3,4] 2 a.__iter__() 3 #列表是可迭代得 4 lst_iterator = a.__iter__() 5 # lst_iterator.__iter__() 6 #lst_iterator就是一个迭代器。 7 print(lst_iterator.__next__()) 8 #执行__next__方法可以遍历里面的数字 9 from collections import Iterator 10 from collections import Iterable 11 print(isinstance(lst_iterator,Iterator)) 12 #判断是否是迭代器 13 print(isinstance(a ,Iterable)) 14 #判断是否是可迭代的
备注:如果迭代器里面的数据遍历完成之后会报错。:stopIteration
异常处理:try,except 可以解决这种问题
1 #异常处理,遍历完成之后会报错,可以用异常处理来解决 2 lis = [1,2,3].__iter__() 3 while True: 4 try: 5 print(lis.__next__()) 6 except StopIteration: 7 #StopIteration ,报错信息。如果提示这个,则直接break 8 break
迭代器的优点:
1、能够对python 中的基本数据类型进行统一的遍历,不需要关心每个值是什么。【例如:字典,你必须知道要他们key才可以取值】
2、它可以节省内存
例如: f = open('file','w')文件句柄就是个天生迭代器 ;range(100) 是个可迭代对象。用.__iter__ 可以把他变成一个迭代器
生成器 (Gerator),Iterator(迭代器):生成器就是迭代器,生成器是自己写出来的
1 def generator_func(): 2 print('123') 3 yield 1111 4 g = generator_func()#---生成器的本质就是迭代器 5 print(g) 6 ret = g.__next__() 7 print(ret) 8 #带yield关键字的函数就是生成器函数 9 #生成器函数在执行的时候返回一个生成器 10 #输出:<generator object generator_func at 0x000002897D1F1B88> 11 #123 12 #1111
# 错误示范:不能这么操作g = generator_func().__next__()
# 生成器可以强转成列表等 例:list(g)
迭代器是可迭代对象。迭代器 = 可迭代对象.___iter__()
从生成器里取值:
1、__inter__ 有几个yield就可以取几次
2、for 喜欢取值
3、注意:再调用生成器函数的时候,要先获取生成器,再进行next 取值
4、其他数据类型进行强转,里面装的是生成器所有的值
5、生成器中的内容只能取一次,按顺序取值,去完为止
6、生成器中的yield一般都有2个以上。如果一个,就相当于return .
7、 yield 一般都是和循环一起使用
例:取衣服,用上循环。
1 def get_clothing(): 2 for cloth in range(100): 3 yield ' 第%s件衣服'%cloth 4 generate = get_clothing() 5 print(next(generate))#==generate.__next__() 6 print(next(generate)) 7 print(next(generate)) 8 for i in range(50):#一次性取50件衣服 9 print(generate.__next__())
# for i in get_clothing():
if i == '第100件衣服':
break
send命令:
1 def func(): 2 value = yield 1 3 yield value 4 g = func() 5 print(g.__next__())# 生成器的第一步只能先用next。 6 print(g.send(10))#到yield位置后,如果有赋值 = 才可以用 7 #send命令,不然就算send也没用! 8 #这里send是赋值给value 9 #send 不仅next的功能,也可以传进去值(赋值操作) 10 #yield 是返回值
备注:如果send 要多次传值必须要有一个未被返回的yield。
如果value没有接收的值的话,会返回None
1、如果第一个yield 只是为了激活生成器。可以写个装饰器(生成器预激装饰器)
1 def wrapper(func): 2 def inner(*args,**kwargs): 3 ret = func(*args,**kwargs) 4 ret.__next__() 5 return ret 6 return inner 7 @wrapper
面试题:查看输出结果
例1
1 def demo(): 2 for i in range(4): 3 yield i 4 g = demo() 5 g1 = (i for i in g) 6 g2 = (i for i in g1) 7 print(list(g1))# g1里面的值已经被list获取,所有g2是个空列表 8 print(list(g2)) 9 #[0,1,2,3] 10 #[]
例2 注意:生成器在调用的时候才生效。下面这道题,就是当N = 10 就一直是10,而不是1,
1 def add(n,i): 2 return n + i 3 def test(): 4 for i in range(4): 5 yield i 6 g = test()#生成器 7 for n in [1,10]: 8 g = (add(n,i) for i in g) 9 # 相当于 10 #n =1 11 #g = (add(n,i) for i in test())n 12 #n =10 13 #g = (add(n,i) for i in add(1,i) for i in test()) 14 print(list(g)) 15 #此时N =10 16 #g = (add(n,i) for i in add(10,i) for i in [0,1,2,3]) 17 #10+0,10+1,10+2,10+3 18 #g = (add(n,i) for i in 10,11,12,13) 19 #10+10,10+11,10+12,10+13 20 #输出 [20, 21, 22, 23]