Python进阶之推导式

列表推导式

1. 普通情况下生成一个列表

# 1-20的列表数据
alist = []
for a in range(1, 21):
  alist.append(a)
print(alist)
# 结果
# [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20]

2. 使用列表推导式,快速生成一个列表

格式1: [计算公式 for 循环 if 判断]

先执行for循环遍历,拿到单个数据在做if判断,满足if 的数据 才会被存到列表中

# 1-20的列表数据
alist = [a for a in range(1, 21)]
print(alist)
# 结果
# [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20]
# 1-20中3的倍数
blist = [b for b in range(1, 21) if b % 3 == 0]
print(blist)
# 结果
# [3, 6, 9, 12, 15, 18]

格式2: [exp1 if 条件 else exp2 for 循环]

先执行for循环遍历,拿到单个数据在做if判断,满足if条件做exp1处理,否则按照exp2处理,最后统一输出

# 1-20中,如果是3的倍数相反数,其他不变
clist = [-c if c % 3 == 0 else c for c in range(1, 21)]
print(clist)
# 结果
# [1, 2, -3, 4, 5, -6, 7, 8, -9, 10, 11, -12, 13, 14, -15, 16, 17, -18, 19, 20]

字典推导式

adict = {str(i): i * i for i in range(10)}
# 结果
# 0': 0, '1': 1, '2': 4, '3': 9, '4': 16, '5': 25, '6': 36, '7': 49, '8': 64, '9': 81}

集合推导式

aset = {i * 2 for i in range(10)}
# 结果
# {0, 2, 4, 6, 8, 10, 12, 14, 16, 18}