如何提取诊断表上面的元素并将其存储在数组中

Question

我创建了一个矩阵使用：

Matrix = [[0 for x in range(5)] for z in range(5)]

我试图提取diagnol上面的元素并将其存储在一个数组中

例如：

    [0, 0, 0, 1, 1]
    [1, 0, 0, 0, 0]
    [1, 1, 0, 0, 1]
    [0, 1, 1, 0, 0]
    [0, 1, 0, 1, 0]
U=[0,0,1,1,0,0,0,0,1,0]

A=[1,4,9]
  [0,1,2]
  [2,3,6]

U=[4,9,2]

Answer 1

你可以只使用列表理解

from random import randrange

Matrix = [[randrange(10) for x in range(5)] for z in range(5)]
>>>Matrix
[[6, 3, 7, 9, 3], [8, 6, 4, 0, 4], [0, 0, 1, 3, 2], [7, 7, 2, 3, 7], [3, 3, 5, 6, 3]]

[Matrix[i][j] for i in range(0,5) for j in range(i+1,5)]
[3, 7, 9, 3, 4, 0, 4, 3, 2, 7]