julia - 从 CSC 矩阵中删除列的最佳方法

Question

至少我有以下稀疏矩阵:

A = sparse([1 4 7; 2 5 8; 3 6 9])

什么是最有效的获取方式:

A = sparse([1 0 7; 2 0 8; 3 0 9])

我使用的实际矩阵实际上要大很多......

Answer 1

要将第二列设置为零，您可以使用:

function set2zero!(A, col)
    @assert 1 <= col <= size(A, 2)
    A.nzval[A.colptr[col]:A.colptr[col+1]-1] .= zero(eltype(A))
    return A
end

这会很快(特别是如果 A 是稀疏的)。

但是，请注意 0 条目存储在矩阵中(因此需要权衡更新速度与以后使用此类数组的速度)。