推導式

把一個迴圈壓成一行。 · ⏱ 約 30 分鐘

你會學到三件事

1. [... for ... in ...] 一行做出一個新的 list
2. 加上 if 就是篩選
3. 什麼時候不該用它

開始之前

這個形狀在 Python 裡到處都是：

squares = []
for n in nums:
    squares.append(n * n)

四行，而它只做一件事：「把每一個都換成別的」。

🟢 先把編輯器打開，跟著下面一起做：跟著做

一、換成別的

一行把一串換成另一串：

1squares = [n * n for n in nums]

從右邊往左邊讀：

[  n * n     for n in nums  ]
   ↑             ↑
 每一個變成    走過 nums 的每一個 n

nums 是 [1,2,3,4,5] 的話，squares 就是 [1,4,9,16,25]。

⚠️ 它做出一個新的 list，不會改到原來的。

二、加上篩選

後面加一個 if，就只留下合的：

1evens = [n for n in nums if n % 2 == 0]

if 放在最後，意思是「只留下符合的」→ [2, 4]。

兩件事可以一起做：

1[n * n for n in nums if n % 2 == 0]     # [4, 16]

讀法：「走過每一個 n，如果是偶數，就換成 n * n」。

三、dict 也可以

換成大括號就是一本字典：

1table = {n: n * n for n in nums}2# {1: 1, 2: 4, 3: 9, 4: 16, 5: 25}

差別只在大括號和那個冒號——和第 4 課分辨 dict / set 的方式一樣。

四、⚠️ 什麼時候不該用

推導式讀起來像一句英文，而那只在它短的時候成立。

下面這個寫法不要學——它塞了太多件事：

1# 🔴 不要這樣2r = [f(x) for row in grid for x in row if g(x) and h(x) or k(x)]

判準：

推導式是「把一串變成另一串」的寫法。 它不是「用比較少的字寫迴圈」的寫法。

完成的樣子

1nums = [1, 2, 3, 4, 5]2squares = [n * n for n in nums]3print(squares)4evens = [n for n in nums if n % 2 == 0]5print(evens)6table = {n: n * n for n in nums}7print(table)

換你了

給一串名字，做出「每個名字的長度」的 list。

再做一個進階的：只留下長度大於 3 的名字。

到編輯器做這一題：練習：每個名字的長度，以及只留長度大於 3 的

做一個

做出 1 到 20 的平方，印出有幾個；再篩出其中的偶數，印出那一串。

🔴 兩行都要是推導式。做完之後用 for 迴圈寫一次同樣的東西， 然後問自己：哪一份比較好讀？而它的答案跟行數沒有關係。

到編輯器做這一題：做一個：平方數裡的偶數

這一課你做了什麼

• 你用一行做出了一個轉換過的 list
• 你用 if 在推導式裡篩選
• 你知道它什麼時候會變成不好讀

如果卡住了