如何通过索引来对矩阵进行子集

Question

A = [1 2 3; 4 5 6] 
r_idx = [1 2] 
c_idx = [1 2] 
A(r_idx,c_idx) = [1 nan nan; nan 5 nan] 

换句话说，它应该返回A的（1,1）和（2,2）元素并将所有其他项设置为nan。

这可能吗？

Answer 1

A = [1 2 3; 4 5 6]; 
r_idx = [1 2]; 
c_idx = [1 2]; 
B = nan(size(A)); 
B(sub2ind(size(A),r_idx,c_idx)) = A(sub2ind(size(A),r_idx,c_idx)); 

然后，期望的结果是在乙

Answer 2

一种方法与sparse -

A(full(sparse(r_idx,c_idx,1,size(A,1),size(A,2)))==0) = nan 

或用setdiff -

A(setdiff(1:numel(A),sub2ind(size(A),r_idx,c_idx))) = nan