在朱莉娅的稀疏矩阵中有效地插入对角线

Question

我正在研究这个资源的EM计算方法。这些方法使用大量大的稀疏矩阵，只有少数对角线设置为非零。所以我的问题是：如何在 Julia 中有效地设置现有矩阵的对角线？

Answer 1

您可以只使用索引分配：

julia> using SparseArrays, LinearAlgebra
julia> S = spzeros(10,10)
10×10 SparseMatrixCSC{Float64,Int64} with 0 stored entries
julia> S[diagind(S)] = rand(10); S
10×10 SparseMatrixCSC{Float64,Int64} with 10 stored entries:
  [1 ,  1]  =  0.2907
  [2 ,  2]  =  0.451863
  [3 ,  3]  =  0.920742
  [4 ,  4]  =  0.0674684
  [5 ,  5]  =  0.587077
  [6 ,  6]  =  0.61916
  [7 ,  7]  =  0.450401
  [8 ,  8]  =  0.596222
  [9 ,  9]  =  0.597324
  [10, 10]  =  0.210721

Answer 2

似乎有一个函数fillband!用值填充两个对角线之间的空间，但由于某种原因，它没有从模块中导出。