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