如何将对角线提取到列向量中？

Question

假设矩阵M:

1 2 3
3 5 6
6 8 9

如何存储我从中提取以下行向量a？

1
5
9

Answer 1

你只需要使用diag:

octave-3.4.0:1> A = [ 1 2 3; 3 5 6; 6 8 9 ]
A =

   1   2   3
   3   5   6
   6   8   9

octave-3.4.0:2> D = diag(A)
D =

   1
   5
   9

请注意,您还可以通过传递第二个参数来提取其他对角线diag,例如

octave-3.4.0:3> D = diag(A, 1)
D =

   2
   6

octave-3.4.0:4> D = diag(A, -1)
D =

   3
   8