julia - 下三角矩阵等于 Julia 中的值

Question

我试图在 Julia 中找到一个下三角矩阵并将该矩阵放入零矩阵(替换零)。然后我想设置下三角矩阵等于值 c。在 R 中，这看起来像这样:

mat <- matrix(0, n,n)
mat[lower.tri(mat, diag=FALSE)] <- c

输出如下所示:

          [,1] [,2]
[1,] 0.0000000    0
[2,] 0.4646787    0

我可以在 Julia 中制作零点矩阵，但我不知道如何将下三角函数与 c 值一起放在其中。

Answer 1

这可能是 Julia 中最接近的等价物:

julia> n = 3; c = 0.4646787; A = zeros(n,n);

julia> A[tril!(trues(size(A)), -1)] .= c;

julia> A
3×3 Array{Float64,2}:
 0.0       0.0       0.0
 0.464679  0.0       0.0
 0.464679  0.464679  0.0

当然，Julia 的一大优势是您几乎不需要担心创建这些巧妙的单行代码。写出朴素的 for 循环非常简单，应该有类似的性能:

julia> for j in 1:size(A, 2), i in j+1:size(A, 1)
           A[i, j] = c
       end

julia> A
3×3 Array{Float64,2}:
 0.0       0.0       0.0
 0.464679  0.0       0.0
 0.464679  0.464679  0.0