matlab - 比较两个不等长的向量得到一个逻辑数组

Question

我需要向量化以下代码:

a = [1 2 3 2 3 1];
b = [1 2 3];

for i = 1:length(a)
    for j = 1:length(b)
        r(i, j) = (a(i) == b(j));
    end
end

输出 r 应该是一个逻辑数组:

 1     0     0
 0     1     0
 0     0     1
 0     1     0
 0     0     1
 1     0     0

我能得到的最接近的是:

for j = 1:length(b)
    r(:, j) = (a == b(j));
end

迭代较短的向量显然更有效，因为它生成的迭代次数更少。正确的解决方案应该没有任何 for 循环。

在 MATLAB/Octave 中这可能吗？

Answer 1

这是一个使用 bsxfun 的简单解决方案.

bsxfun(@eq,b,a')

ans =

     1     0     0
     0     1     0
     0     0     1
     0     1     0
     0     0     1
     1     0     0