Matrix indexing using a vector

Fabio Pasquarella

2016 5 30

2 个回答

回答已采纳

2 次查看（30 天）

0 个投票

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]

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

请先登录，再回答此问题。

Follow Question

采纳的回答

Stephen23 2016-5-30

编辑：Stephen23 2016-5-30

在 MATLAB Online 中打开

1 个投票

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