一、前言

列表生成式即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語言進階學習」，可以通過以下二維碼關注。轉載本文請聯系Go語言進階學習公眾號。