为了得到领先的对角线，你可以做
diag = [ mat[i][i] for i in range(len(mat)) ]
甚至
diag = [ row[i] for i,row in enumerate(mat) ]
并为其他对角线玩类似的游戏。例如，对于反对角线（从右上到左下），您可以执行以下操作：
diag = [ row[-i-1] for i,row in enumerate(mat) ]
对于其他较小的对角线，您必须
if在列表推导中使用条件，例如：
diag = [ row[i+offset] for i,row in enumerate(mat) if 0 <= i+offset < len(row)]
