列表推導(List comprehension)的作用是為了更方便地生成列表(list)。
比如,一個list變量的元素均為數字,如果需要將每個元素的值乘以2并生成另外一個list,下面是一種做法:
代碼如下:
#-*-encoding:utf-8-*-
list1 = [1,2,4,5,12]
list2 = []
for item in list1:
list2.append(item*2)
print list2
如果使用列表推導,可以這樣:
代碼如下:
#-*-encoding:utf-8-*-
list1 = [1,2,4,5,12]
list2 = [item*2 for item in list1 ]
print list2
可以通過if過濾掉不想要的元素,例如提取出list1中小于10的元素:
代碼如下:
#-*-encoding:utf-8-*-
list1 = [1,2,4,5,12]
list2 = [item for item in list1 if item < 10 ]
print list2
如果要將兩個list中的元素進行組合,可以:
代碼如下:
#-*-encoding:utf-8-*-
list1 = [1,2,3]
list2 = [4,5,6]
list3 = [(item1,item2) for item1 in list1 for item2 in list2 ]
print list3
官方文檔中給出了一個比較復雜的轉置矩陣的例子:
代碼如下:
#-*-encoding:utf-8-*-
matrix1 = [
[1, 2, 3, 4],
[5, 6, 7, 8],
[9, 10, 11, 12]
]
matrix2 = [[row[i] for row in matrix1] for i in range(4)]
for row in matrix2:
print row
運行結果如下:
代碼如下:
[1, 5, 9]
[2, 6, 10]
[3, 7, 11]
[4, 8, 12]
新聞熱點
疑難解答