在 Julia 中将多个参数作为一个参数传递

Question

我正在尝试在 Julia 中制作对角块矩阵。我有一个nxn数组，我想将其P复制为对角线上的块矩阵，而矩阵的其余部分是稀疏的。

如果arr是我的阵列，我知道我可以使用：

blockdiag(sparse(arr),sparse(arr))

创建一个块，其中 P=2 对角线上的数组副本。

但是，对于大 P，我如何使用变量以一般方式执行此操作P？

我尝试制作一个 ( nxnxP)数组，但BlockDiag()不接受 3D 数组。

Answer 1

fill可以重复元素而无需实际制作副本。因此你可以这样做：

blockdiag(fill(sparse(arr), 2)...)

这是一个完整的 Julia 会话：

julia> using SparseArrays

julia> arr=Matrix(sprand(Float64,3,3,0.25))
3×3 Matrix{Float64}:
 0.0       0.016897  0.0
 0.219705  0.0       0.0
 0.0       0.0       0.893547


julia> blockdiag(fill(sparse(arr), 2)...)
6×6 SparseMatrixCSC{Float64, Int64} with 6 stored entries:
  ?        0.016897   ?         ?         ?         ?
 0.219705   ?         ?         ?         ?         ?
  ?         ?        0.893547   ?         ?         ?
  ?         ?         ?         ?        0.016897   ?
  ?         ?         ?        0.219705   ?         ?
  ?         ?         ?         ?         ?        0.893547