Python 的列表推導式，這個看似簡單的語法糖，實則內含無限威力。在 Python 程式碼編寫中，列表推導式的靈活性和簡潔性讓它成為了不可或缺的一部分。在這篇文章中，我們將更全面、更深入地探討列表推導式，從基礎的概念認識，到各類進階的用法和操作，我們一一揭祕。最後，我們還將在 "One More Thing" 部分分享一個非常有趣且實用的列表推導式技巧，這會讓你在程式設計道路上又多一份強大的工具。
列表推導式，就是一種在 Python 中建立列表的方式，它的基礎形式如下：
[expression for item in iterable]
它實質上是一個 for 迴圈的簡化形式。例如，我們可以用它來建立一個包含 0 到 9 平方的列表：
squares = [x**2 for x in range(10)]
print(squares) # Output: [0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
這樣，你不需要再使用傳統的 for 迴圈來建立列表，程式碼更加簡潔和清晰。
列表推導式更強大的地方在於，我們可以在其中加入條件判斷，以過濾出我們想要的元素：
even_squares = [x**2 for x in range(10) if x % 2 == 0]
print(even_squares) # Output: [0, 4, 16, 36, 64]
在這個例子中，我們只生成了偶數的平方，只需加入一個簡單的 if 條件，我們就可以靈活地過濾出我們需要的元素。
更進一步，列表推導式還可以巢狀使用，處理更復雜的資料結構，比如我們要將一個巢狀列表展平：
nested_list = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
flattened_list = [x for sublist in nested_list for x in sublist]
print(flattened_list) # Output: [1, 2, 3, 4, 5, 6, 7, 8, 9]
這個例子中，我們將一個二維的巢狀列表展平成了一維列表，就像是將多層次的資料展開，方便我們進行後續處理。
列表推導式中的表示式可以幫助我們對資料進行變形：
strings = ['Hello', 'World', 'In', 'Python']
lowercase_strings = [s.lower() for s in strings]
print(lowercase_strings) # Output: ['hello', 'world', 'in', 'python']
在這個例子中，我們將一個包含幾個字串的列表，通過 str.lower() 函數，將其轉換成了全小寫。通過改變表示式，我們可以在生成新列表的同時，對資料進行各種變形操作。
推導式不僅僅可以應用於列表，還可以推廣到字典和集合中：
squared_dict = {x: x**2 for x in range(5)}
print(squared_dict) # Output: {0: 0, 1: 1, 2: 4, 3: 9, 4: 16}
squared_set = {x**2 for x in range(5)}
print(squared_set) # Output: {0, 1, 4, 16, 9}
這兩個例子分別演示了字典推導式和集合推導式的用法，這些結構的推導式可以更方便我們處理複雜的資料結構。
列表推導式不僅程式碼更加簡潔，實際上在很多情況下，列表推導式的執行效率也優於傳統的 map 或 filter 函數：
import time
# Using list comprehension
start_time = time.time()
squares = [x**2 for x in range(1000000)]
end_time = time.time()
print(f"List comprehension took {end_time - start_time} seconds")
# Using map function
start_time = time.time()
squares = list(map(lambda x: x**2, range(1000000)))
end_time = time.time()
print(f"Map function took {end_time - start_time} seconds")
在這個例子中，我們分別用列表推導式和 map 函數建立一個包含一百萬個元素的列表，可以看到列表推導式的執行時間通常要少於 map 函數。
在我的 GitHub 學習過程以及在各種技術部落格中閱讀，我發現一個關於列表推導式的有趣且實用的技巧，那就是使用列表推導式實現全排列：
perms = [(x, y, z) for x in range(3) for y in range(3) for z in range(3) if x != y and y != z and x != z]
print(perms) # Output: [(0, 1, 2), (0, 2, 1), (1, 0, 2), (1, 2, 0), (2, 0, 1), (2, 1, 0)]
這個例子通過列表推導式生成了 3 個元素的全排列，使用了巢狀迴圈和條件判斷，非常簡潔而高效。
總的來說，Python 的列表推導式是一個非常強大而靈活的工具，能夠幫助我們更好地處理資料和建立資料結構。希望這篇文章能幫助你更深入地理解和應用列表推導式，讓你的 Python 程式碼更加簡潔和高效。
如有幫助，請多關注
個人微信公眾號：【Python全視角】
TeahLead_KrisChang，10+年的網際網路和人工智慧從業經驗，10年+技術和業務團隊管理經驗，同濟軟體工程本科，復旦工程管理碩士，阿里雲認證雲服務資深架構師，上億營收AI產品業務負責人。