我有一个 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(??? ???)
最佳答案
我建议使用相应的 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]
关于python - 在python中打印矩阵的右对角线值,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/36560854/