matlab - 在 MATLAB 中将零分配给矩阵的特定索引

Question

例如:

a = [1 2 3; 4 5 6; 7 8 9];   
b = [2 4]; %//Indices I got

如何将 a 中未在 b 中索引的每个元素设置为零以获得:

0 2 0  
4 0 0   
0 0 0

我尝试了for循环:

for i = 1:numel(a)  
    if i ~= b  
      a(i) = 0;
    end       
end

但是我处理的矩阵非常大，并且需要很长的时间才能完成运行。

有什么聪明的方法吗？谢谢。

Answer 1

试试这个:

a = [1 2 3; 4 5 6; 7 8 9];
b = [2 4]; 

a(setdiff(1:length(a(:)),b)) = 0;

更新

正如@Daniel建议的，对于大矩阵最好使用

a(setdiff(1:numel(a),b)) = 0;