MATLAB - 矢量化包含距離度量的雙循環

Question

我想優化我的代碼，我不知道如何以及如果我能夠矢量化這個特定的部分？

for base_num = 1:base_length 
    for sub_num = 1:base_length 
    dist{base_num}(sub_num) = sqrt((x(base_num) - x(sub_num))^2 + (y(base_num) - y(sub_num))^2); 
    end 
end 

Answer 1

以下示例提供矢量化的一個方法：

%# Set example parameters 
N = 10; 
X = randn(N, 1); 
Y = randn(N, 1); 

%# Your loop based solution 
Dist1 = cell(N, 1);  
for n = 1:N 
    for m = 1:N 
     Dist1{n}(m) = sqrt((X(n) - X(m))^2 + (Y(n) - Y(m))^2); 
    end 
end 

%# My vectorized solution 
Dist2 = sqrt(bsxfun(@minus, X, X').^2 + bsxfun(@minus, Y, Y').^2); 
Dist2Cell = num2cell(Dist2, 2); 

快速速度測試在N = 1000具有運行兩個數量級比循環溶液更快矢量化的解決方案。

注意：我在向量化解決方案中使用了第二行來模仿您的單元格數組輸出結構。由你決定是否要包括它或兩個結合成一條線等。

順便說一句，在問題中發佈代碼+1。但是，對於未來的兩點小建議：1）發佈到SO時，使用簡單的變量名 - 尤其是循環下標 - 例如我的答案。 2）當我們可以將示例代碼直接複製並粘貼到腳本中並運行而無需進行任何更改或添加（如在我的答案中）時，這很好。這使我們能夠更快地融合解決方案。