Tworzenie zerowej wypełnioną tablicę 2D z nich w pozycjach indeksowanych przez wektor

Question

Próbuję wektorować następującą operację MATLAB:

Biorąc wektor kolumnowy z indeksami, chcę matrycę z same liczba wierszy kolumny i stała liczba kolumn. Macierz jest inicjowana zerami i zawiera te w lokalizacji określone przez indeksy.

Oto przykład skryptu już napisałem:

y = [1; 3; 2; 1; 3]; 
m = size(y, 1); 

% For loop 
yvec = zeros(m, 3); 
for i=1:m 
    yvec(i, y(i)) = 1; 
end 

Pożądany wynik jest:

yvec = 

1  0  0 
0  0  1 
0  1  0 
1  0  0 
0  0  1 

Czy to możliwe, aby osiągnąć ten sam efekt bez pętli for ? Próbowałem coś takiego:

% Vectorization (?) 
yvec2 = zeros(m, 3); 
yvec2(:, y(:)) = 1; 

ale to nie działa.

Answer 1

Dwa podejścia, których można użyć tutaj.

Podejście 1:

y = [1; 3; 2; 1; 3]; 
yvec = zeros(numel(y),3); 
yvec(sub2ind(size(yvec),1:numel(y),y'))=1 

Podejście 2 (One-liner):

yvec = bsxfun(@eq, 1:3,y) 

Answer 2

Jeszcze inne podejście:

yvec = full(sparse(1:numel(y),y,1)); 

Answer 3

Można to zrobić z accumarray :

yvec = accumarray([(1:numel(y)).' y], 1); 

Answer 4

zrobiłem to w ten sposób:

classes_count = 10; 
sample_count = 20; 
y = randi([1 classes_count], 1, sample_count); 

y_onehot = zeros(classes_count, size(y, 2)); 
idx = sub2ind(size(y_onehot), y, [1:size(y, 2)]); 
y_onehot(idx) = 1