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

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));

然后，期望的结果在B

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