Python推导式编程,第1张

推导式comprehensions(又称解析式),是Python的一种独有特性。推导式是可以从一个数据序列构建另一个新的数据序列的结构体, 共有三种推导,在Python2和3中都有支持:

◆ 列表(list)推导式

◆ 字典(dict)推导式

◆ 集合(set)推导式

一、列表推导式

列表推导式书写形式:

[表达式 for 变量 in 列表] 或者 [表达式 for 变量 in 列表 if 条件]

1.无条件判断条件的推导式

例如:我们有一个列表list1,现在需要将list1中的每个元素都乘以2,并添加到一个新列表里,然后打印这个新列表。

先看常规for循环的写法:

list1 = [1, 2, 3, 4, 5, 6, 7]new_list = []for i in list1: i = i * 2 new_list.append(i)print(new_list) # [2, 4, 6, 8, 10, 12, 14]

列表推导式写法:

# 列表推导式list1 = [1, 2, 3, 4, 5, 6, 7]new_list = [i * 2 for i in list1]  # 无判断条件的推导式print(new_list)  # [2, 4, 6, 8, 10, 12, 14]

可以看到,采用列表推导式后,代码量就由6行缩减至了3行

2.有条件判断的推导式

但是在实际工作中,我们遇到的情况可能会更复杂,比如涉及到条件判断。在列表推导式中,也是可以写if语句的。

同样地,先看常规for循环的写法:

list1 = [1, 2, 3, 4, 5, 6, 7]new_list = []for i in list1: if i == 2: i = i * 2 new_list.append(i)print(new_list) # [4]

列表推导式写法:

# 有判断条件的推导式list1 = [1, 2, 3, 4, 5, 6, 7]num = [i * 2 for i in list1 if i == 2]print(num)  # [4]

可以看到,采用列表推导式后,代码量就由7行缩减至了3行。

3.多层循环嵌套的推导式

列表推导式并没有限制循环层数,多层循环就是一层一层的嵌套,你可以展开一个两层或三层的列表推导式,例如我们上一篇文章在介绍for循环时提到的利用双层for循环打印“九九乘法表”,

普通的for循环写法:

# 九九乘法表for i in range(1, 10): for j in range(1, i + 1): print('{} * {} = {}'.format(j, i, i * j), end=' ') print()

这里我们用列表推导式来实现一下:

print('\n'.join([' '.join([f'{j} * {i} = {j * i}' for j in range(1, i + 1)]) for i in range(1, 10)]))

打印效果如下:

Python推导式编程,文章图片1,第2张

可以看出,尽管代码缩减到了一行就能够实现打印九九乘法表,但带来的问题同样显而易见:可读性差!时间久了,可能我们自己都忘记这段代码的逻辑。所以,过于复杂的场景、尤其是多层for循环嵌套不推荐使用列表推导式进行实现。

二、字典推导式

字典推导和列表推导的使用方法是类似的,只把中括号该改成大括号。

# 字典推导式dict1 = {'name': 'chen', 'age': 28, 'num': 227}new_dict = {k: v for k, v in dict1.items()}print(new_dict) # {'name': 'chen', 'age': 28, 'num': 227}new_dict_reverse = {v: k for k, v in dict1.items()} # 字典键值互换位置print(new_dict_reverse) # {'chen': 'name', 28: 'age', 227: 'num'}

三、集合推导式

集合推导式跟列表推导式差不多,都是对一个列表的元素全部执行相同的操作,但集合是一种无重复无序的序列

跟列表推到式的区别在于:

  • 不使用中括号,使用大括号;
  • 结果中无重复;
  • 结果是一个set()集合;

例如我们相对集合中的每一个元素都乘以元素本身,即求平方,然后生成一个新的集合。

先来看一下常规的for循环写法:

set1 = {1, 2, 3, 4, 5, 6, 7}new_set = set({})for i in set1:    i = i ** 2    new_set.add(i)print(new_set)

集合推导式写法:

# 集合推导式set1 = {1, 2, 3, 4, 5, 6, 7}new_set = {i ** 2 for i in set1} # 遍历集合的元素,返回元素的平方print(new_set) # {1, 4, 36, 9, 16, 49, 25}

总结

推导式常用于转换数据、过滤数据等场景,并不是所有的场景都适合使用推导式,在追求代码运行效率的同时,也要关注代码可读性,而不是一味追求“逼格”。

DABAN RP主题是一个优秀的主题,极致后台体验,无插件,集成会员系统
白度搜_经验知识百科全书 » Python推导式编程

0条评论

发表评论

提供最优质的资源集合

立即查看 了解详情