Julia 稀疏矩阵

Question

我有向量 y_vec，如何将向量转换为 Y_matrix

形式的矩阵

y_vec = [0; 1; 1; 2; 3; 4]

Y_matrix = [1 0 0 0 0
            0 1 0 0 0
            0 1 0 0 0
            0 0 1 0 0
            0 0 0 1 0
            0 0 0 0 1]

到目前为止，我已经尝试使用 for 循环。

Y_mat = full(spzeros(length(y_vec), length(unique(y_vec))))

for (i,j) in enumerate(1:length(y_vec))
    Y_mat[i, y_vec[j]+1] = 1
end

但是，当y_vec不连续时似乎有问题，比如y_vec = [0; 1; 1; 2; 3; 4; 8]，使用for循环失败！！！如何解决这个问题。

有没有办法在 Julia 中使用稀疏矩阵来解决上述问题

Answer 1

你可以使用稀疏矩阵构造函数sparse(I,J,V):

y_vec = [0; 1; 1; 2; 3; 4; 8]
I = collect(1:length(y_vec))
J = y_vec+1
V = ones(length(y_vec))
S = sparse(I,J,V)
full(S)

julia> full(S)
7x9 Array{Float64,2}:
 1.0  0.0  0.0  0.0  0.0  0.0  0.0  0.0  0.0
 0.0  1.0  0.0  0.0  0.0  0.0  0.0  0.0  0.0
 0.0  1.0  0.0  0.0  0.0  0.0  0.0  0.0  0.0
 0.0  0.0  1.0  0.0  0.0  0.0  0.0  0.0  0.0
 0.0  0.0  0.0  1.0  0.0  0.0  0.0  0.0  0.0
 0.0  0.0  0.0  0.0  1.0  0.0  0.0  0.0  0.0
 0.0  0.0  0.0  0.0  0.0  0.0  0.0  0.0  1.0

Julia 稀疏矩阵

Julia sparse matrix

matrix

sparse-matrix

julia