matlab - 矢量化 Octave /matlab代码

Question

以下是 Octave 编码(kmeans的一部分)

centroidSum = zeros(K);
valueSum = zeros(K, n);
for i = 1 : m  
  for j = 1 : K    
    if(idx(i) == j)
      centroidSum(j) = centroidSum(j) + 1;
      valueSum(j, :) = valueSum(j, :) + X(i, :);
    end
  end
end

代码有效，是否可以对代码进行矢量化？无需 if 语句即可轻松对代码进行矢量化，但是我们如何使用 if 语句对代码进行矢量化呢？

Answer 1

我假设代码的目的是计算 n 维空间中一组 m 数据点的子集的质心，这些点存储在该空间中在矩阵 X(点 x 坐标)和向量 idx 中为每个数据点指定子集 (1 ... K) 该点属于。那么一个部分向量化是:

centroid = zeros(K, n)
for j = 1 : K
   centroid(j, :) = mean(X(idx == j, :));
end

if 通过索引消除，特别是逻辑索引:idx == j 给出一个 bool 数组，指示哪些数据点属于子集 j.

我认为也可以去掉第二个 for 循环，但这会导致代码非常复杂、难以理解。