快速（稀疏）矩阵在MATLAB

Question

我填充与基于功能f一些值稀疏矩阵生成使用嵌套的循环，像这样：

B = sparse(x, y); 
for ix=1:x 
    for jy=1:y 
     if f(ix,jy) 
       B(ix, jy) = 1; 
     end 
    end 
end 

这是做事的最佳和最快的方式matlab，还是有一个优化的方式？

Answer 1

你可以尝试像以下，这就避免了循环：

如果

[is, js] = ind2sub(size(B), 1:numel(B)); 
fval = arrayfun(f, is, js); 
B(find(fval)) = 1; 

不知道这将是虽然更快。

如果f接受向量而不是单个索引，则可跳过arrayfun。

Answer 2

如果你的函数可以接受向量输入，您可以计算的f所有值在一杆，然后用sparse：

[II,JJ] = ndgrid(1:y,1:x); 
F = reshape(f(II(:),JJ(:)),x,y); % maybe just f(II,JJ) if f allows it 
[ii,jj] = find(F); 
B = sparse(ii,jj,1,x,y); 

如果你的函数不接受向量，你将不得不通过运行它该循环或使用arrayfun来生成F(ix,iy) = f(ix,iy);，然后如上所述使用find和sparse。