matrix - 如何从 Julia 中的对角矩阵中提取对角元素数组？

Question

给定 Julia 中的一些对角矩阵，例如 A = Diagonal(rand(3,3))，有什么方法可以创建一个一维数组，其元素是此 Diagonal 的对角元素矩阵 A?

Answer 1

diag(A, k::Integer=0) 返回矩阵 A 的第 k 条对角线作为向量。

julia> A = Diagonal(rand(3,3))
3×3 Diagonal{Float64, Vector{Float64}}:
 0.213159   ⋅         ⋅
  ⋅        0.034186   ⋅
  ⋅         ⋅        0.539693

julia> diag(A)
3-element Vector{Float64}:
 0.21315894297089488
 0.03418604147090787
 0.5396925608269262