python - 在python中打印矩阵的右对角线值

Question

我有一个 5 x 5 整数矩阵，需要在 Python 中使用某些代码。我需要构建一个列表，其中包含矩阵从右上角到左下角的对角线上的单元格中的值。

matrix = [[ 1,  2,  3,  4,  5],
          [ 6,  7,  8,  9, 10],
          [11, 12, 13, 14, 15],
          [16, 17, 18, 19, 20],
          [21, 22, 23, 24, 25]]

期望的输出:

[5, 9, 13, 17, 21]

部分代码尝试:

diagonal = []
for posi in matrix???:
    diagonal.append(??? ???)

Answer 1

我建议使用相应的 numpy 类，在本例中为 numpy.matrix。然后，Numpy 允许我们从左向右翻转值并提取对角线。

import numpy as np

mymatrix = np.matrix([[ 1,  2,  3,  4,  5],
          [ 6,  7,  8,  9, 10],
          [11, 12, 13, 14, 15],
          [16, 17, 18, 19, 20],
          [21, 22, 23, 24, 25]])

mydiagonal = np.matrix.diagonal(np.fliplr(mymatrix))

print(mydiagonal)
[[ 5  9 13 17 21]]

如果输出必须是列表，则可以使用 tolist() 轻松转换:

print(mydiagonal.tolist()[0])
[5, 9, 13, 17, 21]