Создайте матрицу с единицами в определенных столбцах

Question

y - вектор-столбец с целыми значениями обозначают номер метки (например, между 1-3). Теперь мне нужно создать матрицу с таким же числом строк y. Каждая строка i теперь представляет собой вектор с элементом y(i), который должен быть 1, а остальные - 0.

Пример:

y = [3, 2, 2]'; 

x = [0,0,1; 
    0,1,0; 
    0,1,0]; 

Есть ли эффективный способ, чтобы закончить эту работу. Я знаю, что цикл будет делать:

x = zeros(size(y, 1), 3) 
for i = 1:size(y, 1) 
    x(i, y(i)) = 1 
end 

Answer 1

Вы можете легко сделать это с помощью sub2ind, который преобразует индексы линейных индексов. Нижние индексы строк: 1:numel(y), а ваши индексы столбца - значения y.

x = zeros(numel(y), max(y)); 
x(sub2ind(size(x), 1:numel(y), y(:).')) = 1; 

Или вы можете просто сделать это с небольшой математикой индекса.

x = zeros(numel(y), max(y)); 
x(((y(:).' - 1) * numel(y)) + (1:numel(y))) = 1;