设置特定的行和列，除了零MATLAB中的对角线元素

Question

比方说我给一些指标像B = [10 23 32....];

现在可以说我有一个矩阵答：我想要做的是对每个指标的B让说我，我想把A的第i行和第i列设置为0，除了对角线元素A（i，i）（它保持不变）。

我可以通过循环做到这一点。但我想要一些基于某种矩阵乘法的矩阵乘法，这比循环更快。

任何想法家伙？

Answer 1

假设，

B=[10 23 32 12 15 18 20] 
M=true(6) 
M(B)=false %making the indexed elements false 
M=or(M,diag(true(size(M,1),1))) %keep the diagonal elements one 

% creating a matrix which has zero in ith row and ith column and diagonal has ones 
M1=and(bsxfun(@or,bsxfun(@and,repmat(min(M,[],2),1,size(M,2)),min(M,[],1)),diag(true(size(M,1),1))),M) 

%Now just multiply M1 with your matrix A, and you are done. 
newA=A.*M1 

可以上面两行合并的一个，但我更喜欢他们脱节为了提高可读性。你有

Answer 2

一种选择是：

创建对角线元素的线性指标：

[I, J]=size(A); 
idx=sub2ind([I,J], B, B); 

设置水平线和垂直为0，更换对角线元素：

NewA=A; 
NewA(B, :)=zeros(numel(B),J); 
NewA(:, B)=zeros(I,numel(B)); 
NewA(idx)=A(idx); 

Answer 3

您可以将对角线元素临时存储在其他位置，将其索引到A中用B将对应的行和列设置为零，并最终插回到对角线元素中 -

%// rows in A 
rows = size(A,1); 

%// Store the diagonal elements temporarily somewhere else 
tmp_diagA = A(1:rows+1:end); 

%// Set the ith rows and cols (obtained from B) to zero 
A(B,:)=0; 
A(:,B)=0; 

%// Plug back in the diagonal elements in place 
A(1:rows+1:end) = tmp_diagA; 

函数调用在MATLAB中应该是昂贵的，我们在这段代码中几乎没有函数调用，所以我希望它足够快。

Answer 4

对于方A：

b = zeros(size(A,1),1); 
b(B) = B; 
A = A.*bsxfun(@eq, b, b.') 

对于一般A：

b1 = zeros(size(A,1),1); 
b1(B) = B; 
b2 = zeros(1,size(A,2)); 
b2(B) = B; 
A = A.*bsxfun(@eq, b1, b2);