Update matrix using vector of column index

Question

Georgio Mazarin 2017-3-25

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/331912-update-matrix-using-vector-of-column-index

评论： Georgio Mazarin 2017-3-25

I have a 2-dim matrix like this:

3436    0.3283    0.3890    0.0274
1137    0.4443    0.3484    0.0889
2748    0.3124    0.4074    0.0439

and a vector with column indexes for each row

     3
     2
     2

Task is to invert values to get result like this:

3436    0.3283   -0.3890    0.0274
1137   -0.4443    0.3484    0.0889
2748   -0.3124    0.4074    0.0439

I believe there is a simple way to do that efficiently without for-loop cycles. Any idea?

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

请先登录，再进行评论。

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

Answer 1

the cyclist 2017-3-25

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/331912-update-matrix-using-vector-of-column-index#answer_260343

编辑：the cyclist 2017-3-25

在 MATLAB Online 中打开

Looks like a job for linear indexing:

M = [0.3436    0.3283    0.3890    0.0274
     0.1137    0.4443    0.3484    0.0889
     0.2748    0.3124    0.4074    0.0439];
v = [3 2 2];
[m,n] = size(M);
linearIndex = sub2ind([m n],1:m,v);
M(linearIndex) = -M(linearIndex)

Note that the 2nd and 3rd arguments to sub2ind have to be the same shape, so make sure they are both column vectors, or both row vectors, but not mixed.