在不使用numpy的情况下获得对角线？

在不使用numpy的情况下获得对角线？

为了得到领先的对角线，你可以做

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)]