所谓列表推导,就是按照一定规则迭代原列表中的元素,并产生一个新的列表。能产生新的列表这也是使用列表推导的一个重要原因。
它的格式:
[表达式 for 变量 in 列表] 或者 [表达式 for 变量 in 列表 if 条件]
列表推导的写法很Pythonic,所以能用列表推导实现的,就不要用循环。如:
list1 = [1,2,4,5,12]
list2 = [item*2 for item in list1 ]
如果要满足某个特定条件,比如过滤掉元素值小于10的元素:
list1 = [1,2,4,5,12]
list2 = [item for item in list1 if item < 10 ]
当然也可以将两个列表合并:
list1 = [1,2,3]
list2 = [4,5,6]
list3 = [(item1,item2) for item1 in list1 for item2 in list2 ]
这里的两个for 是独立的,分别迭代。
再来个复杂点的,转置矩阵:
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
这里的两个for 类似于 for 嵌套,因为使用了 [] 将一个 for 包裹起来,相当于是外面的 for 的内层循环。
[1, 5, 9]
[2, 6, 10]
[3, 7, 11]
[4, 8, 12]