推导式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= [] foriinlist1: i=i*2new_list.append(i) print(new_list) # [2, 4, 6, 8, 10, 12, 14]
列表推导式写法:
# 列表推导式list1= [1, 2, 3, 4, 5, 6, 7] new_list= [i*2foriinlist1] # 无判断条件的推导式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= [] foriinlist1: ifi==2: i=i*2new_list.append(i) print(new_list) # [4]
列表推导式写法:
# 有判断条件的推导式list1= [1, 2, 3, 4, 5, 6, 7] num= [i*2foriinlist1ifi==2] print(num) # [4]
可以看到,采用列表推导式后,代码量就由7行缩减至了3行。
3.多层循环嵌套的推导式
列表推导式并没有限制循环层数,多层循环就是一层一层的嵌套,你可以展开一个两层或三层的列表推导式,例如我们上一篇文章在介绍for循环时提到的利用双层for循环打印“九九乘法表”,
普通的for循环写法:
# 九九乘法表foriinrange(1, 10): forjinrange(1, i+1): print("{} * {} = {}".format(j, i, i*j), end=" ") print()
这里我们用列表推导式来实现一下:
print('\n'.join([' '.join([f"{j} * {i} = {j*i}"forjinrange(1, i+1)]) foriinrange(1, 10)]))
打印效果如下:
可以看出,尽管代码缩减到了一行就能够实现打印九九乘法表,但带来的问题同样显而易见:可读性差!时间久了,可能我们自己都忘记这段代码的逻辑。所以,过于复杂的场景、尤其是多层for循环嵌套不推荐使用列表推导式进行实现。
二、字典推导式
字典推导和列表推导的使用方法是类似的,只把中括号该改成大括号。
# 字典推导式dict1= {"name": "chen", "age": 28, "num": 227} new_dict= {k: vfork, vindict1.items()} print(new_dict) # {'name': 'chen', 'age': 28, 'num': 227}new_dict_reverse= {v: kfork, vindict1.items()} # 字典键值互换位置print(new_dict_reverse) # {'chen': 'name', 28: 'age', 227: 'num'}
三、集合推导式
集合推导式跟列表推导式差不多,都是对一个列表的元素全部执行相同的操作,但集合是一种无重复无序的序列
跟列表推到式的区别在于:
- 不使用中括号,使用大括号;
- 结果中无重复;
- 结果是一个set()集合;
例如我们相对集合中的每一个元素都乘以元素本身,即求平方,然后生成一个新的集合。
先来看一下常规的for循环写法:
set1= {1, 2, 3, 4, 5, 6, 7} new_set=set({}) foriinset1: i=i**2new_set.add(i) print(new_set)
集合推导式写法:
# 集合推导式set1= {1, 2, 3, 4, 5, 6, 7} new_set= {i**2foriinset1} # 遍历集合的元素,返回元素的平方print(new_set) # {1, 4, 36, 9, 16, 49, 25}
总结
推导式常用于转换数据、过滤数据等场景,并不是所有的场景都适合使用推导式,在追求代码运行效率的同时,也要关注代码可读性,而不是一味追求“逼格”。