How to operate on specific diagonal elements of a matrix?

2 次查看(过去 30 天)
Hello everyone.
I am having a matrix k with size (20 x 20), for which, I need to replace some of the diagonal elements with ones. I constructed a vector v including the indices of the elements to be replaced, and applied the following code, but I is not working as I wanted.
k = rand(20);
u = [1; 3; 6; 11; 16; 20];
k(u(:,1),u(:,1)) = 1;
He is replacing the diagonal members and other members as if he is including all combinations of the rows and columns. How can I edit this to get only the diagonal members corrected?

采纳的回答

Stephen23
Stephen23 2017-11-27
编辑:Stephen23 2017-11-27
You need to use sub2ind:
>> idx = sub2ind(size(k),u,u);
>> k(idx) = 1

更多回答(0 个)

类别

Help CenterFile Exchange 中查找有关 Operating on Diagonal Matrices 的更多信息

产品

Community Treasure Hunt

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

Start Hunting!

Translated by