Matrix indexing using a vector

1 view (last 30 days)
Let's say I have a vector b where each row correspond to an index for each row in the matrix A. I want to assign 1 to these elements, like in the matrix C, without loops. Thank you
A = zeros(4,3);
b = [2;1;2;3];
C = [0 1 0; 1 0 0; 0 1 0; 0 0 1]

Accepted Answer

Stephen23
Stephen23 on 30 May 2016
Edited: Stephen23 on 30 May 2016
You can use sub2ind for this:
>> A = zeros(4,3);
>> col = [2;1;2;3];
>> row = 1:numel(col);
>> A(sub2ind(size(A),row(:),col(:))) = 1
A =
0 1 0
1 0 0
0 1 0
0 0 1

More Answers (1)

Andrei Bobrov
Andrei Bobrov on 30 May 2016
C = accumarray([(1:numel(b))' b],1)

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!