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}