向量化逻辑以检查索引是否匹配

时间:2017-05-08 04:21:18

标签: matlab math octave vectorization

我有以下功能完美,但我想将矢量化应用于它......

for i = 1:size(centroids,1)
    centroids(i, :) = mean(X(idx == i, :));
end

它检查idx是否与当前索引匹配,如果匹配,则会计算与该索引对应的所有mean值的X值。

这是我对矢量化的尝试,我的解决方案不起作用,我知道为什么......

centroids = mean(X(idx == [1:size(centroids,1)], :));

以下idx == [1:size(centroids,1)]会破坏代码。我不知道如何检查idx是否等于从1size(centroids,1)的任何一个数字。

TL:博士

通过矢量化摆脱for循环

0 个答案:

没有答案