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

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

Answer 2

bsxfun(@eq, a', b)