matlab代码嵌套循环性能提升

Question

我很想收到有关如何提高以下嵌套 for 循环性能的建议：

I = (U > q); % matrix of indicator variables, I(i,j) is 1 if U(i,j) > q

for i = 2:K

    for j = 1:(i-1)

        mTau(i,j) = sum(I(:,i) .* I(:,j));

        mTau(j,i) = mTau(i,j);

    end

end

代码评估变量对是否两个变量都低于某个阈值，从而填充矩阵。感谢您的帮助！

Answer 1

你可以使用矩阵乘法：

I = double(U>q);
mTau = I.'*I;

这将在对角线上有非零值，因此您可以将它们设置为零

mTau = mTau - diag(diag(mTau));

Answer 2

bsxfun 的一种方法-

out = squeeze(sum(bsxfun(@and,I,permute(I,[1 3 2])),1));
out(1:size(out,1)+1:end)=0;