1.推导式的基本概念
推导式是可以从一个数据序列构建另一个新的数据序列的结构体。
推导式有三种形式:
(1)列表推导式 (list)
(2)字典推导式 (dict)
(3)集合推导式 (set)
2.列表推导式
列表推导能非常简洁的构造一个新的列表:只能用一条简洁的表达式就可以对得到的元素进行转换。
案例1:#求(x,y)其中x是0-5之间的偶数,y是0-5之间的奇数组成的元祖列表。
print([(x,y) for x in range(5) if x%2==0 for y in range(5) if y %2==1])
输出结果:[(0, 1), (0, 3), (2, 1), (2, 3), (4, 1), (4, 3)]
案例2:#求m中3,6,9组成的列表。
m = [[1,2,3],[4,5,6],[7,8,9]]
print([m[x][2] for x in (0,1,2)])
输出结果:[3, 6, 9]
案例3:#求斜线1,5,9组成的列表。
m=[[1,2,3],[4,5,6],[7,8,9]]
print([m[x][x] for x in range(len(m))])
案例4:#求m,n中矩阵和元素的乘积
m=[[1,2,3],[4,5,6],[7,8,9]]
n=[[2,2,2],[3,3,3],[4,4,4]]
print([m[x][y]*n[x][y]for x in range(3)for y in range(3)])
案例5:#结合两个列表的元素,如果元素之间不相等的话
combs=[]
for x in [1,2,3]:
for y in [3,1,4]:
if x !=y:
combs.append((x,y))
print([(x, y) for x in [1,2,3] for y in [3,1,4] if x != y])
3.字典推导式
字典和集合推导式的语法差不多,只不过产生的是集合和字典。
案例:#快速更换key和value
mcase={'a':10,'b':34}
print({k:v for v,k in mcase.items()})
4.集合推导式
集合推导式跟列表推导式很像,唯一区别在于用{}代替[]。
案例1:
squared = {x**2 for x in [1, 1, 2]}
print(squared)
输出结果:{1, 4}