列表推导式的公式

[Expression for target in iterable]

循环和列表推导式的转换.png

使用案例

推导式使用demo.png

用列表推导式找出对角线上的数字：

matrix = [[1,2,3],
          [4,5,6],
          [7,8,9]]
# 打印出中间一列
# 中间列在矩阵中的坐标为（0，1）（1，1）（2，1）
a = [matrix[i][1] for i in range(len(matrix))]
a = [2,5,8]
# 打印出两个对角线
# 左起对角线上的数字在矩阵中的坐标为（0，0）（1，1）（2，2）
a = [matrix[i][i] for i in range(len(matrix))]
a = [1,5,9]
# 打印出两个对角线
# 右起对角线上的数字在矩阵中的坐标为（0，2）（1，1）（2，0）
a = [matrix[i][len(matrix)-1-i] for i in range(len(matrix))]
a = [3,5,7]