写点什么

软件测试|Python 实用炫酷技能——推导式

  • 2023-05-14
    北京
  • 本文字数:1539 字

    阅读完需:约 5 分钟

Python 推导式

判断一个程序员水平的高低,不能光看他的发量,也不能光看他的代码量,还要看他代码蕴含的思想,代码的质量。代码蕴含的思想主要体现在各种设计模式的运用上,而代码的质量就既要实现需求,又要保证代码的简洁优雅。保证代码质量是需要长期积累,养成良好的编程习惯,不断思考优化的。


今天就给大家介绍一种实用的保证代码简洁的武功—— Python 推导式。

什么是推导式

推导式 comprehensions(又称解析式),是 python 的一种独有特性。推导式是可以从一个数据序列构建另一个新的数据序列。


看推导式的定义很抽象,下面我们来看看具体的例子,通过例子来学习常用的四种推导式。

列表推导式

我们先来看一个需求:


快速创建一个包含元素 1-9 的平方的列表


面对这个需求,我们通常的实现方式是这样的:


list = []for i in range(1, 10):    list.append(i*i)print(list) # [1, 4, 9, 16, 25, 36, 49, 64, 81]
复制代码


但是如果我们使用列表推导式,只需要一行代码就可以实现:


lis = [x * x for x in range(1, 10)]print(list) # [1, 4, 9, 16, 25, 36, 49, 64, 81]
复制代码


查看代码结构,我们可以发现:


变量名 = [表达式 for 变量 in 列表]
复制代码


更复杂一些的例子:


list = [x * y for x in range(1, 10) for y in range(1, 10)]print(list)# [1, 2, 3, 4, 5, 6, 7, 8, 9, 2, 4, 6, 8, 10, 12, 14, 16, 18, 3, 6, 9, 12, 15, 18, 21, 24, 27, 4, 8, 12, 16, 20, 24, 28, 32, 36, 5, 10, 15, 20, 25, 30, 35, 40, 45, 6, 12, 18, 24, 30, 36, 42, 48, 54, 7, 14, 21, 28, 35, 42, 49, 56, 63, 8, 16, 24, 32, 40, 48, 56, 64, 72, 9, 18, 27, 36, 45, 54, 63, 72, 81]
复制代码


这个是多个变量的类型,因此我们可以扩展上面的列表推导式:


变量名 = [表达式 for 变量 in 列表 for 变量 in xxx]
复制代码


列表推导式还可以加上条件,变成这样


变量名 = [表达式 for 变量 in 列表 if 条件]
复制代码


例如,快速创建一个包含 1-100 之间所有偶数的列表。


list = [i for i in range(1, 101) if i % 2 == 0]print(list) # [2, 4, 6, 8, 10, 12, 14, 16, 18, 20, 22, 24, 26, 28, 30, 32, 34, 36, 38, 40, 42, 44, 46, 48, 50, 52, 54, 56, 58, 60, 62, 64, 66, 68, 70, 72, 74, 76, 78, 80, 82, 84, 86, 88, 90, 92, 94, 96, 98, 100]
复制代码

字典推导式

字典推导式的形式和列表推导式类似,语法也是类似的,只不过字典推导式返回的结果是字典。


变量名 = {key: value 表达式}


来看一个实例,


dict = {x: x/2 for x in range(1,11) if x % 2 == 0}print(dict)# 输出: {2: 1.0, 4: 2.0, 6: 3.0, 8: 4.0, 10: 5.0}
复制代码

集合推导式

集合推导式与列表和字典类似,所以这里直接写出来:


变量名 = {表达式 for 变量 in 列表 for 变量 in xxx}
复制代码


或者带有条件:


变量名 = {表达式 for 变量 in 列表 if 条件}
复制代码


实际应用,生成一个存储 10 以内偶数的集合


set = {x for x in range(10) if x % 2 == 0}print(set)
# 输出 {0, 2, 4, 6, 8}
复制代码

元组推导式

按照上述 3 个推导式的思路,我们先试着通过推导式输出一个元组,代码如下:


tup=(x for x in range(1,10))print(tup)
# 输出: <generator object <genexpr> at 0x00000221A68C7BC8>
复制代码


并没有如我们预期的那样生成元组,上面的代码返回的变量其实是一个生成器,并不是一个元组。其实没有真正的元组推导式,我们只能用一个类似的方法来生成元组。


上面代码进行一下小改进就可以生成元组了:


tup=tuple(x for x in range(1,10))print(tup)
# 输出: (1, 2, 3, 4, 5, 6, 7, 8, 9)
复制代码


在推导式前,加上 tuple 就可以达到我们的目标。

总结

这里介绍了四种 python 的推导式,主要是用来简化 循环的代码,生成不同的数据结构用的。当然,从这些基础的表达式语法出发,还可以运用到复杂的推导式,大家在写代码时留心使用就会发现它的强大之处。在面试时,也是一个炫技加分项!


用户头像

社区:ceshiren.com 微信:ceshiren2021 2019-10-23 加入

微信公众号:霍格沃兹测试开发 提供性能测试、自动化测试、测试开发等资料,实时更新一线互联网大厂测试岗位内推需求,共享测试行业动态及资讯,更可零距离接触众多业内大佬。

评论

发布
暂无评论
软件测试|Python实用炫酷技能——推导式_霍格沃兹测试开发学社_InfoQ写作社区