有效地将矢量解包为二进制矩阵Octave

Question

在Octave我试图以以下格式解压缩矢量:

y = [ 1  
      2  
      4  
      1 
      3 ]

我想返回一个维度矩阵(rows(y)x max value(y)),其中对于每一行,我在原始数字值的列中有一个1,在其他地方有一个零,即上面的例子

y01 = [ 1 0 0 0
        0 1 0 0
        0 0 0 1
        1 0 0 0
        0 0 1 0 ]

到目前为止我有

y01 = zeros( m, num_labels );
for i = 1:m
    for j = 1:num_labels
        y01(i,j) = (y(i) == j);
    end
end

哪个有效,但是对于更大的矩阵来说会变得缓慢,并且看起来效率低下,因为即使大多数没有变化,它也会循环遍历每一个值.

我在另一个线程上为R找到了这个:

f3 <- function(vec) {
    U <- sort(unique(vec))
    M <- matrix(0, nrow = length(vec), 
          ncol = length(U), 
          dimnames = list(NULL, U))
    M[cbind(seq_len(length(vec)), match(vec, U))] <- 1L
    M
}

但我不知道R和我不确定解决方案是否/如何端口到八度音程.

谢谢你的任何建议!

Answer 1

使用稀疏矩阵(这也节省了大量内存),可以像往常一样用于进一步的计算:

y = [1; 2; 4; 1; 3]
y01 = sparse (1:rows (y), y, 1)

如果你真的想要一个完整的矩阵,那么使用"full":

full (y01)
ans =
1   0   0   0
0   1   0   0
0   0   0   1
1   0   0   0
0   0   1   0