假设我有一个NxN矩阵A,一个由数字1:N的子集和一个值K组成的索引向量V,我想这样做:
for i = V
A(i,i) = K
end
有没有办法在一个带矢量化的声明中做到这一点?
e.g。 A(某事)= K
语句A(V,V) = K
不起作用,它分配非对角元素,这不是我想要的。 e.g:
>> A = zeros(5);
>> V = [1 3 4];
>> A(V,V) = 1
A =
1 0 1 1 0
0 0 0 0 0
1 0 1 1 0
1 0 1 1 0
0 0 0 0 0
答案 0 :(得分:59)
我通常使用EYE:
A = magic(4)
A(logical(eye(size(A)))) = 99
A =
99 2 3 13
5 99 10 8
9 7 99 12
4 14 15 99
或者,你可以创建线性索引列表,因为从一个对角元素到下一个对角元素,需要nRows+1
步:
[nRows,nCols] = size(A);
A(1:(nRows+1):nRows*nCols) = 101
A =
101 2 3 13
5 101 10 8
9 7 101 12
4 14 15 101
如果您只想访问对角元素的子集,则需要创建对角线索引列表:
subsetIdx = [1 3];
diagonalIdx = (subsetIdx-1) * (nRows + 1) + 1;
A(diagonalIdx) = 203
A =
203 2 3 13
5 101 10 8
9 7 203 12
4 14 15 101
或者,您可以使用diag
创建逻辑索引数组(仅适用于方阵)
diagonalIdx = false(nRows,1);
diagonalIdx(subsetIdx) = true;
A(diag(diagonalIdx)) = -1
A =
-1 2 3 13
5 101 10 8
9 7 -1 12
4 14 15 101
答案 1 :(得分:23)
>> tt = zeros(5,5)
tt =
0 0 0 0 0
0 0 0 0 0
0 0 0 0 0
0 0 0 0 0
0 0 0 0 0
>> tt(1:6:end) = 3
tt =
3 0 0 0 0
0 3 0 0 0
0 0 3 0 0
0 0 0 3 0
0 0 0 0 3
更一般:
>> V=[1 2 5]; N=5;
>> tt = zeros(N,N);
>> tt((N+1)*(V-1)+1) = 3
tt =
3 0 0 0 0
0 3 0 0 0
0 0 0 0 0
0 0 0 0 0
0 0 0 0 3
这是基于矩阵可以作为一维数组(向量)访问的事实,其中2个索引(m,n)被线性映射m * N + n替换。
答案 2 :(得分:2)
A = zeros(7,6);
V = [1 3 5];
[n m] = size(A);
diagIdx = 1:n+1:n*m;
A( diagIdx(V) ) = 1
A =
1 0 0 0 0 0
0 0 0 0 0 0
0 0 1 0 0 0
0 0 0 0 0 0
0 0 0 0 1 0
0 0 0 0 0 0
0 0 0 0 0 0
答案 3 :(得分:2)
假设K是值。命令
A=A-diag(K-diag(A))
可能会快一点
>> A=randn(10000,10000);
>> tic;A(logical(eye(size(A))))=12;toc
经过的时间是0.517575秒。
>> tic;A=A+diag((99-diag(A)));toc
经过的时间是0.353408秒。
但它消耗更多内存。
答案 4 :(得分:2)
我使用sub2ind
并将对角线索引作为x和y参数传递:
A = zeros(4)
V=[2 4]
idx = sub2ind(size(A), V,V)
% idx = [6, 16]
A(idx) = 1
% A =
% 0 0 0 0
% 0 1 0 0
% 0 0 0 0
% 0 0 0 1
答案 5 :(得分:2)
>> B=[0,4,4;4,0,4;4,4,0]
B =
0 4 4
4 0 4
4 4 0
>> v=[1,2,3]
v =
1 2 3
>> B(eye(size(B))==1)=v
%insert values from v to eye positions in B
B =
1 4 4
4 2 4
4 4 3
答案 6 :(得分:1)
我在有限差分代码中使用了这个小的内联函数。
A=zeros(6,3);
range=@(A,i)[1-min(i,0):size(A,1)-max(i+size(A,1)-size(A,2),0 ) ];
Diag=@(A,i) sub2ind(size(A), range(A,i),range(A,i)+i );
A(Diag(A, 0))= 10; %set diagonal
A(Diag(A, 1))= 20; %equivelent to diag(A,1)=20;
A(Diag(A,-1))=-20; %equivelent to diag(A,-1)=-20;
通过更改功能范围,可以轻松修改它以对角线的子范围工作。