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

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)