最新下载
热门教程
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
Python全栈之推导式和生成器代码示例
时间:2022-06-25 01:32:55 编辑:袖梨 来源:一聚教程网
本篇文章小编给大家分享一下Python全栈之推导式和生成器代码示例,文章代码介绍的很详细,小编觉得挺不错的,现在分享给大家供大家参考,有需要的小伙伴们可以来看看。
1. 推导式
# ### 推导式 : 通过一行循环判断遍历出一些列数据的方法叫做推导式 """ 语法: val for val in iterable """ # 1.推导式基本语法 lst = [] for i in range(1,51): lst.append(i) print(lst) # 改写推导式 lst = [ i for i in range(1,51) ] print(lst) # 小练习 # 1.[1,2,3,4,5] => [2,4,6,8,10] lst = [ i*2 for i in range(1,6) ] print(lst) # 2.带有判断条件的推导式 """注意点:for后面紧跟的判断条件只能是单项分支.""" """[1,2,3,4,5,6,7,8,9,10] => [1,3,5,7,9 ... ]""" lst = [1,2,3,4,5,6,7,8,9,10] lst_new = [] for i in lst: if i % 2 == 1: lst_new.append(i) print(lst_new) # 改写推导式 lst = [ i for i in lst if i % 2 == 1 ] print(lst) # 3.多循环推导式 # 谁♡♢♤♠谁 lst1 = ["孙杰龙","陈露","曹静怡"] lst2 = ["王志国","邓鹏","合理"] lst_new = [] for i in lst1: for j in lst2: lst_new.append(i+"♡♢♤♠"+j) print(lst_new) # 改写推导式 lst = [ i+"♡♢♤♠"+j for i in lst1 for j in lst2 ] print(lst) # 4.带有判断条件的多循环推导式 lst_new = [] for i in lst1: for j in lst2: if lst1.index(i) == lst2.index(j): lst_new.append(i+"♡♢♤♠"+j) print(lst_new) # 改写推导式 lst = [i+"♡♢♤♠"+j for i in lst1 for j in lst2 if lst1.index(i) == lst2.index(j)] print(lst)
2. 推导式练习
# ### 推导式练习题 # (1).{'x': 'A', 'y': 'B', 'z': 'C' } 把字典写成x=A,y=B,z=C的列表推导式 dic = {'x': 'A', 'y': 'B', 'z': 'C' } lst = [] for k,v in dic.items(): res = k + "=" + v lst.append(res) print(lst) # 推导式 lst = [ k + "=" + v for k,v in dic.items() ] print(lst) # (2).把列表中所有字符变成小写 ["ADDD","dddDD","DDaa","sss"] lst = ["ADDD","dddDD","DDaa","sss"] lst_new = [] for i in lst: lst_new.append(i.lower()) print(lst_new) # 推导式 lst = [ i.lower() for i in lst ] print(lst) # (3).x是0-5之间的偶数,y是0-5之间的奇数 把x,y组成一起变成元组,放到列表当中 # 方法一 lst = [] for x in range(6): for y in range(6): if x % 2 == 0 and y % 2 == 1: lst.append( (x,y) ) print(lst) # 推导式 lst = [ (x,y) for x in range(6) for y in range(6) if x % 2 == 0 and y % 2 == 1 ] print(lst) # 方法二 lst = [] for x in range(6): if x % 2 == 0 : for y in range(6): if y % 2 == 1: lst.append( (x,y) ) print(lst) # 推导式 lst = [ (x,y) for x in range(6) if x % 2 == 0 for y in range(6) if y % 2 == 1 ] print(lst) # (4).使用列表推导式 制作所有99乘法表中的运算 for i in range(1,10): for j in range(1,i+1): print("{:d}*{:d}={:2d} ".format(i,j,i*j) , end="") print() lst = ["{:d}*{:d}={:2d} ".format(i,j,i*j) for i in range(1,10) for j in range(1,i+1)] print(lst) # (5)求M,N中矩阵和元素的乘积 # M = [ [1,2,3], # [4,5,6], # [7,8,9] ] # N = [ [2,2,2], # [3,3,3], # [4,4,4] ] M = [ [1,2,3] ,[4,5,6] , [7,8,9] ] N = [ [2,2,2] ,[3,3,3] , [4,4,4] ] """ M[0][0] * N[0][0] = 2 M[0][1] * N[0][1] = 4 M[0][2] * N[0][2] = 6 M[1][0] * N[1][0] = 12 M[1][1] * N[1][1] = 15 M[1][2] * N[1][2] = 18 M[2][0] * N[2][0] = 12 M[2][1] * N[2][1] = 15 M[2][2] * N[2][2] = 18 """ # =>实现效果1 [2, 4, 6, 12, 15, 18, 28, 32, 36] """双层循环,外层循环动的慢,内层循环动的快,正好符号M N 矩阵的下标""" lst = [] for i in range(3): for j in range(3): lst.append( M[i][j] * N[i][j] ) print(lst) # =>实现效果2 [ [2, 4, 6], [12, 15, 18], [28, 32, 36] ] # 遍历出三个空的列表 lst = [ [] for i in range(3)] print(lst) lst = [ [ M[i][j] * N[i][j] for j in range(3) ] for i in range(3)] print(lst) """ [ M[i][j] * N[i][j] for j in range(3) ] [ [2, 4, 6] [12, 15, 18] [28, 32, 36] ] """
3. 集合_字典推导式
# ### 集合推导式 """ 案例: 满足年龄在18到21,存款大于等于5000 小于等于5500的人, 开卡格式为:尊贵VIP卡老x(姓氏),否则开卡格式为:抠脚大汉卡老x(姓氏) 把开卡的种类统计出来 """ lst = [ {"name":"赵沈阳","age":18,"money":3000}, {"name":"赵万里","age":19,"money":5200}, {"name":"赵蜂拥","age":20,"money":100000}, {"name":"赵世超","age":21,"money":1000}, {"name":"王志国","age":18,"money":5500}, {"name":"王永飞","age":99,"money":5500} ] setvar = set() for i in lst: print(i) # {'name': '赵沈阳', 'age': 18, 'money': 3000} if 18 <= i["age"] <= 21 and 5000 <= i["money"] <= 5500: res = "尊贵VIP卡老{}".format(i["name"][0]) else: res = "抠脚大汉卡老{}".format(i["name"][0]) # 添加到集合中 setvar.add(res) print(setvar) # { 三元运算符 + 推导式 } # 三运运算符 + 推导式 setvar = { "尊贵VIP卡老{}".format(i["name"][0]) if 18 <= i["age"] <= 21 and 5000 <= i["money"] <= 5500 else "抠脚大汉卡老{}".format(i["name"][0]) for i in lst } print(setvar) # ### 字典推导式 ### 一.enumerate """ enumerate(iterable,[start=0]) 功能:枚举 ; 将索引号和iterable中的值,一个一个拿出来配对组成元组,通过迭代器返回 参数: iterable: 可迭代性数据 (常用:迭代器,容器类型数据,可迭代对象range) start: 可以选择开始的索引号(默认从0开始索引) 返回值:迭代器 """ # 基本语法 from collections import Iterator,Iterable lst =["王文","吕洞宾","何仙姑","铁拐李","张国老","曹国舅","蓝采和","韩湘子"] it = enumerate(lst) it = enumerate(lst,start=100) print(isinstance(it,Iterator)) # next print( next(it) ) # for + next (推荐,数据较大时使用) for i in range(3): print(next(it)) # for for i in it: print(i) # list 强转迭代器 print(list(it)) # (1) 字典推导式 配合 enumerate 来实现 dic = {k:v for k,v in enumerate(lst,start=100)} print(dic) """ (100, '王文') (101, '吕洞宾') (102, '何仙姑') (103, '铁拐李') (104, '张国老') (105, '曹国舅') (106, '蓝采和') (107, '韩湘子') """ # (2) 使用dict强转迭代器,瞬间得到字典 dic = dict( enumerate(lst,start=100) ) print(dic) ### 二.zip """ 特点:按照索引配对 zip(iterable, ... ...) 功能: 将多个iterable中的值,一个一个拿出来配对组成元组,通过迭代器返回 iterable: 可迭代性数据 (常用:迭代器,容器类型数据,可迭代对象range) 返回: 迭代器 """ # 基本语法 # lst1 = ["孙开启","王永飞","于朝志"] # lst2 = ["薛宇健","韩瑞晓","上朝气"] # lst3 = ["刘文博","历史园","张光旭"] # 在索引下标同时存在时,才会进行配对,否则舍弃. lst1 = ["孙开启","王永飞","于朝志"] lst2 = ["薛宇健","韩瑞晓"] lst3 = ["刘文博"] it = zip(lst1,lst2,lst3) print(list(it)) # (1) 字典推导式 配合 zip 来实现 lst_key = ["ww","axd","yyt"] lst_val = ["王维","安晓东","杨元涛"] # ('ww', '王维'), ('axd', '安晓东'), ('yyt', '杨元涛') dic = {k:v for k,v in zip(lst_key , lst_val) } print(dic) # (2) 使用dict强转迭代器,瞬间得到字典 dic = dict( zip(lst_key , lst_val) ) print(dic)
4. 生成器
4.1 生成器表达式
# ### 生成器 """ #生成器本质是迭代器,允许自定义逻辑的迭代器 #迭代器和生成器区别: 迭代器本身是系统内置的.重写不了. 而生成器是用户自定义的,可以重写迭代逻辑 #生成器可以用两种方式创建: (1)生成器表达式 (里面是推导式,外面用圆括号) (2)生成器函数 (用def定义,里面含有yield) """ # (1) 生成器表达式 (里面是推导式,外面用圆括号) gen = ( i for i in range(10) ) print(gen) # 判断类型 from collections import Iterator,Iterable print(isinstance(gen,Iterator)) # 1.next 调用生成器 print(next(gen)) print(next(gen)) # 2.for + next 调用生成器 for i in range(3): print(next(gen)) # 3.for 调用生成器所有数据 for i in gen: print(i) # 4.list强转生成器,瞬间得到所有数据 gen = ( i for i in range(10) ) print(list(gen)) # print(next(gen)) error # StopIteration
4.2 生成器函数
# ### 生成器函数 """ # yield 类似于 return 共同点在于:执行到这句话都会把值返回出去 不同点在于:yield每次返回时,会记住上次离开时执行的位置 , 下次在调用生成器 , 会从上次执行的位置往下走 而return直接终止函数,每次重头调用. yield 6 和 yield(6) 2种写法都可以 yield 6 更像 return 6 的写法 推荐使用 """ # (1) 基本语法 def mygen(): print("111") yield 1 print("222") yield 2 print("333") yield 3 # 初始化生成器函数 => 返回生成器对象 => 简称生成器 gen = mygen() # 第一次调用 res = next(gen) print(res) # 第二次调用 res = next(gen) print(res) # 第三次调用 res = next(gen) print(res) # 第四次调用 """ StopIteration error res = next(gen) print(res) """ """ # 第一次调用 print("111") yield 1 保存当前第13行代码的状态,把1返回,并且等待下一次调用 # 第二次调用 从上一次保存的位置13行往下走, print("222") yield 2 保存当前第16行代码的状态,把2返回,并且等待下一次调用 # 第三次调用 从上一次保存的位置16行往下走, print("333") yield 3 保存当前第19行代码的状态,把3返回,并且等待下一次调用 # 第四次调用 因为没有更多的yield 返回数据,所有停止迭代.出现报错异常. """ # (2) 优化生成器代码 """生成器应用的场景是在大数据的范围中使用,切记不可直接用for遍历所有,可能无法短时间内获取所有数据""" def mygen(): for i in range(1,101): yield i # 初始化生成器函数 => 生成器 gen = mygen() print("<=====>") for i in range(30): num = next(gen) print("我的球衣号码是{}".format(num)) print("<=====>") for i in range(40): num = next(gen) print("我的球衣号码是{}".format(num)) # (3) send的使用方式 (给上一个yield发送数据) """ # next和send区别: next 只能取值 send 不但能取值,还能发送值 # send注意点: 第一个 send 不能给 yield 传值 默认只能写None 最后一个yield 接受不到send的发送值 """ def mygen(): print("start") res = yield "内部1" print(res,"<==内部==>") res = yield "内部2" print(res,"<==内部==>") res = yield "内部3" print(res,"<==内部==>") print("end") # 初始化生成器函数 => 生成器 gen = mygen() # 第一次调用生成器 """ 第一次调用生成器时,因为没有遇到yield保存的代码位置, 无法发送数据,默认第一次只能发送None """ res = gen.send(None) print(res,"<==外部==>") # 第二次调用生成器 res = gen.send("100") print(res,"<==外部==>") # 第三次调用生成器 res = gen.send("200") print(res,"<==外部==>") # 第四次调用生成器 """ error res = gen.send("300") print(res,"<==外部==>") """ """ 使用send调用生成器,第一次发送时必须是None,因为还没有遇到yield保存的代码位置 res = gen.send(None) 走到mygen生成器函数中 print("start") res = yield "内部1" 执行第80行 ,保存退出,记录当前代码位置,将 "内部1" 返回 在98行接受数据 res = "内部1" print(内部1,"<==外部==>") 第二次调用生成器 res = gen.send("100") 把100这个数据发送给上一次代码保存的位置80行进行接受. => 导致 80行 res = 100 打印81行 print(100 ,"<==内部==>") 执行83行 res = yield "内部2" 保存退出,记录当前代码位置,将 "内部2" 返回 执行102行 res = gen.send("100") => "内部2" print("内部2","<==外部==>") .... 依次类推 ... 到第四次调用时, 因为没有更多的yield 返回数据,gen.send(300)无法接受到返回值,所以出现停止迭代 StopIteration的报错,程序终止; """ # (4) yield from 的使用 """将一个可迭代对象变成一个迭代器返回 """ def mygen(): lst = ["张磊","李亚峰","刘一峰","王同培"] yield from lst # 初始化生成器函数 gen = mygen() print(next(gen)) print(next(gen)) print(next(gen)) print(next(gen)) # print(next(gen)) # StopIteration # (5) 斐波那契数列 """使用生成器分段获取所有内容,而不是一股脑的把所有数据全部打印""" """1 1 2 3 5 8 13 21 34 .... """ def mygen(maxval): a,b = 0,1 i = 0 while i < maxval: # print(b) yield b a,b = b,a+b i += 1 # mygen(10) gen = mygen(10) # 第一次获取 for i in range(3): print(next(gen)) # 第二次获取 for i in range(5): print(next(gen))
5. 小练习
# 1.用推导式写如下程序 # (1)构建如下列表:[0, 2, 4, 6, 8, 10, 12, 14, 16, 18] lst = [i * 2 for i in range(10)] lst = [i for i in range(0,19,2) ] print(lst) # (2)lst = ['alex', 'WuSir', '老男孩', '神秘男孩'] 将lst构建如下列表:['alex0', 'WuSir1', '老男孩2', '神秘男孩3'] lst = ['alex', 'WuSir', '老男孩', '神秘男孩'] # 方法一 # lst = [i + str(lst.index(i)) for i in lst] # 方法二 lst = [lst[i] + str(i) for i in range(len(lst)) ] print(lst) # (3)构建如下列表:[(0, 1), (1, 2), (2, 3), (3, 4), (4, 5), (5, 6)] lst = [ (j,i) for j in range(0,6) for i in range(1,7) if i-j == 1] print(lst) lst = [(i,i+1) for i in range(6)] print(lst) # (4)求出50以内能被3整除的数的平方,并放入到一个列表中。 lst = [i ** 2 for i in range(51) if i % 3 == 0] print(lst) # (5)M = [[1,2,3],[4,5,6],[7,8,9]], 把M中3,6,9组成新列表 M = [[1,2,3],[4,5,6],[7,8,9]] lst = [ i[-1] for i in M ] print(lst) # (6)构建如下列表:['python1期', 'python2期', 'python3期', 'python4期', 'python6期', 'python7期', 'python8期', 'python9期', 'python10期'] lst = [ "python{}期".format(i) for i in range(1,11) if i != 5 ] print(lst) # (7)过滤掉长度小于3的字符串列表 , 并转换成大写字母 lst = ["sdfsdfsdfsdf","234","你说的符号是","a","ab"] lst = [ i.upper() for i in lst if len(i) >=3 ] print(lst) # (8)除了大小王,里面有52项,每一项是一个元组,请返回如下扑克牌列表[('红心','2'),('草花','J'), …('黑桃','A')] lst1 = ["红心","草花","黑桃","方片"] lst2 = ["A","2","3","4","5","6","7","8","9","10","J","Q","K"] lst = [(i,j) for i in lst1 for j in lst2] print(lst) # 2.用推导式写如下程序 lst1 = { 'name':'alex', 'Values':[ {'timestamp': 1517991992.94,'values':100,}, {'timestamp': 1517992000.94,'values': 200,}, {'timestamp': 1517992014.94,'values': 300,}, {'timestamp': 1517992744.94,'values': 350}, {'timestamp': 1517992800.94,'values': 280} ] } # 将lst1 转化成如下 lst2: lst2 = [ [1517991992.94, 100], [1517992000.94, 200], [1517992014.94, 300], [1517992744.94, 350], [1517992800.94, 280] ] # 方法一 lst2 = [ [i["timestamp"] , i["values"]] for i in lst1["Values"] ] print(lst2) # 方法二 lst2 = [ list(i.values()) for i in lst1["Values"]] print(lst2) # 3.读取一个文件所有内容,通过生成器调用一次获取一行数据. def mygen(filename): with open(filename,mode="r+",encoding="utf-8") as fp: for i in fp: yield i gen = mygen("ceshi111.txt") for i in range(3): print(next(gen)) for i in range(6): print(next(gen)) # 4.将普通求和函数改写成yield写法 print("<====>") def add(a,b): yield a + b mygen = add(34,5) for i in mygen: print(i)