快速（稀疏）矩陣在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。