盤點Python列表生成式的三種方法

作者：Go進階者 2021-02-06 11:26:55

列表生成式即List Comprehensions，是Python內置的非常簡單卻強大的可以用來創建list的生成式。

一、前言

列表生成式即List Comprehensions，是Python內置的非常簡單卻強大的可以用來創建list的生成式。

要生成list [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]可以用list(range(1, 11))。

print(list(range(1, 11)))

如果要生成[1x1, 2x2, 3x3, …, 10x10]怎么做?

1. 方法一是循環：

L = [] 
for x in range(1, 11): 
    L.append(x * x) 
print(L)

但是循環太繁瑣，而列表生成式則可以用一行語句代替循環生成上面的list：

print([x * x for x in range(1, 11)])

寫列表生成式時，把要生成的元素x * x放到前面，后面跟for循環，就可以把list創建出來，十分有用，多寫幾次，很快就可以熟悉這種語法。

for循環后面還可以加上if判斷，這樣就可以篩選出僅偶數的平方：

for x in range(1, 11): 
    L.append(x * x) 
 
print([x * x for x in range(1, 11) if x % 2 == 0])

2. 使用兩層循環，可以生成全排列

L = [] 
for x in range(1, 11): 
    L.append(x * x) 
 
print( [m + n for m in 'ABC' for n in 'XYZ'])

三層和三層以上的循環就很少用到了。

3. 運用列表生成式，可以寫出非常簡潔的代碼。

例如，列出當前目錄下的所有文件和目錄名，可以通過一行代碼實現：

import os  # 導入os模塊，模塊的概念后面講到 
 
print([d for d in os.listdir('.')])  # os.listdir可以列出文件和目錄

for循環其實可以同時使用兩個甚至多個變量，s比如dict的items()可以同時迭代key和value：

d = {'x': 'A', 'y': 'B', 'z': 'C'} 
for k, v in d.items(): 
    print(k, '=', v)

因此，列表生成式也可以使用兩個變量來生成list：

d = {'x': 'A', 'y': 'B', 'z': 'C' } 
print([k + '=' + v for k, v in d.items()]

最后把一個list中所有的字符串變成小寫：

L = ['Hello', 'World', 'IBM', 'Apple'] 
print([s.lower() for s in L])

如果list中既包含字符串，又包含整數，由于非字符串類型沒有lower()方法，所以列表生成式會報錯：

L = ['Hello', 'World', 18, 'Apple', None] 
print([s.lower() for s in L])

使用內建的isinstance函數可以判斷一個變量是不是字符串：

x = 'abc' 
y = 123 
print(isinstance(x, str)) 
 
print(isinstance(y, str))

請修改列表生成式，通過添加if語句保證列表生成式能正確地執行。

# -*- coding: utf-8 -*- 
L1 = ['Hello', 'World', 18, 'Apple', None] 
L2=??? 
 
# 期待輸出: ['hello', 'world', 'apple'] 
print(L2)

練習參考代碼 :

L2 =[s.lower() for s in L1 if isinstance(s, str) ]

運行結果：

注:

運用列表生成式，可以快速生成list，可以通過一個list推導出另一個list，而代碼卻十分簡潔。

本文基于Python基礎，介紹了列表生成式，以案例為主，三種方法進行講解。

分析不同方法的實現一樣的效果。實際案例中需要注意的點，遇到的難點，提供有效的解決方法。最后通過練習和思考，對列表生成式加深理解。

歡迎大家積極嘗試，有時候看到別人實現起來很簡單，但是到自己動手實現的時候，總會有各種各樣的問題，切勿眼高手低，勤動手，才可以理解的更加深刻。

代碼很簡單，希望對你學習有幫助。

責任編輯：華軒來源： Go語言進階學習