matlab - 按行的交叉索引

鉴于这两个矩阵:

m1 = [ 1 1;
       2 2;
       3 3;
       4 4;
       5 5 ];

m2 = [ 4 2;
       1 1;
       4 4;
       7 5 ];

我正在寻找一个功能，例如:

indices = GetIntersectionIndecies (m1,m2);

其输出将是

indices =
          1
          0
          0
          1
          0

如何在不使用循环 的情况下找到这两个矩阵 之间的行的交集索引？

最佳答案

一种可能的解决方案:

function [Index] = GetIntersectionIndicies(m1, m2)
[~, I1] = intersect(m1, m2, 'rows');
Index = zeros(size(m1, 1), 1);
Index(I1) = 1;

顺便说一句，我喜欢@Shai 的创造性解决方案，如果您的矩阵很小，它比我的解决方案快得多。但是如果您的矩阵很大，那么我的解决方案将占主导地位。这是因为如果我们设置 T = size(m1, 1) ，那么@Shai 的答案中的 tmp 变量将是 T*T，即如果 T 很大，则是一个非常大的矩阵。这是一些用于快速速度测试的代码:

%# Set parameters
T = 1000;
M = 10;

%# Build test matrices
m1 = randi(5, T, 2);
m2 = randi(5, T, 2);

%# My solution
tic
for m = 1:M
[~, I1] = intersect(m1, m2, 'rows');
Index = zeros(size(m1, 1), 1);
Index(I1) = 1;
end
toc

%# @Shai solution
tic
for m = 1:M
tmp = bsxfun( @eq, permute( m1, [ 1 3 2 ] ), permute( m2, [ 3 1 2 ] ) );
tmp = all( tmp, 3 ); % tmp(i,j) is true iff m1(i,:) == m2(j,:)
imdices = any( tmp, 2 );
end
toc

设置 T = 10 和 M = 1000 ，我们得到:

Elapsed time is 0.404726 seconds. %# My solution
Elapsed time is 0.017669 seconds. %# @Shai solution

但是设置 T = 1000 和 M = 100 ，我们得到:

Elapsed time is 0.068831 seconds. %# My solution
Elapsed time is 0.508370 seconds. %# @Shai solution

关于matlab - 按行的交叉索引，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/14495189/