列表推导式的公式
[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]
