Python列表的推导


列表推导表达式(list comprehension expression)在Python中是一个高级的语法操作。
列表推导源自数学中集合的概念。它是一种通过对序列中的每一项运行一个表达式来创建一个新列表的方法。

例如,推导在处理矩阵问题上非常强大。

# 定义一个矩阵
m = [[1, 2, 3],
     [4, 5, 6],
     [7, 8, 9]]
print("矩阵:{}".format(m))
# 使用推导表达式获取矩阵的第二列
# 把矩阵m的每个row中的row[1]防灾一个新的列表中
col2 = [row[1] for row in m]
print("获取矩阵的第二列: {}".format(col2))
矩阵:[[1, 2, 3], [4, 5, 6], [7, 8, 9]]
获取矩阵的第二列: [2, 5, 8]

稍微复杂的操作。

print("矩阵第二列 + 1: {}".format([row[1] + 1 for row in m]))
print("矩阵第二列中的偶数: {}".format([row[1] for row in m if row[1] % 2 == 0]))
矩阵第二列 + 1: [3, 6, 9]
矩阵第二列中的偶数: [2, 8]

注意列表推导是编写在方括号[]中的,提醒是在创建列表。
结合range()函数的例子:

print(list(range(4)))
print(list(range(-6, 7, 2)))
print([[x ** 2, x ** 3] for x in range(4)])
print([[x, x / 2, x * 2] for x in range(-6, 7, 2) if x > 0])
[0, 1, 2, 3]
[-6, -4, -2, 0, 2, 4, 6]
[[0, 0], [1, 1], [4, 8], [9, 27]]
[[2, 1.0, 4], [4, 2.0, 8], [6, 3.0, 12]]