Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Efficiently unpack a vector into binary matrix Octave

On Octave I'm trying to unpack a vector in the format:

y = [ 1  
      2  
      4  
      1 
      3 ]

I want to return a matrix of dimension ( rows(y) x max value(y) ), where for each row I have a 1 in the column of the original digits value, and a zero everywhere else, i.e. for the example above

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

so far I have

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

which works, but is going get slow for bigger matrices, and seems inefficient because it is cycling through every single value even though the majority aren't changing.

I found this for R on another thread:

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
}

but I don't know R and I'm not sure if/how the solution ports to octave.

Thanks for any suggestions!

like image 804
ralph346526 Avatar asked May 28 '15 06:05

ralph346526


1 Answers

Use a sparse matrix (which also saves a lot of memory) which can be used in further calculations as usual:

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

if you really want a full matrix then use "full":

full (y01)
ans =
1   0   0   0
0   1   0   0
0   0   0   1
1   0   0   0
0   0   1   0
like image 187
Andy Avatar answered Sep 21 '22 12:09

Andy