根据第二个矩阵中的值过滤矩阵行

Question

给定一个 2x3 矩阵 x 和一个 4x2 矩阵 y，我想使用 y 的每一行来索引到 x。如果x 中的值不等于-1，我想从y 中删除该行。这是一个可以做我想做的事情的例子，除了我想以一种快速、简单的方式来做，没有循环。

x = [1, 2, 3; -1, 2, -1];
y = [1, 1; 1, 3; 2, 1; 2, 3];

for i=size(y,1):-1:1
   if x(y(i,1), y(i,2)) ~= -1
      y(i,:) = [];
   end
end

这会导致：

y =

     2     1
     2     3

Answer 1

sub2ind 遵循的原始方法（由 Luis 发布的这个漂亮的 solution 使用）本质上就是这样 -

y = y(x((y(:,2)-1)*size(x,1)+y(:,1))==-1,:)

---

基准测试

基准代码

N = 5000;
num_runs = 10000;

x = round(rand(N,N).*2)-1;
y = zeros(N,2);
y(:,1) = randi(size(x,1),N,1);
y(:,2) = randi(size(x,2),N,1);

disp('----------------- With sub2ind ')
tic
for k = 1:num_runs
    y1 = y(x(sub2ind(size(x), y(:,1), y(:,2)))==-1,:);
end
toc,clear y1

disp('----------- With raw version of sub2ind ')
tic
for k = 1:num_runs
    y2 = y(x((y(:,2)-1)*size(x,1)+y(:,1))==-1,:);
end
toc

结果

----------------- With sub2ind 
Elapsed time is 4.095730 seconds.
----------- With raw version of sub2ind 
Elapsed time is 2.405532 seconds.

Answer 2

这可以很容易地向量化如下（参见sub2ind）：

y = y(x(sub2ind(size(x), y(:,1), y(:,2)))==-1,:);

Answer 3

>> x = [1, 2, 3; -1, 2, -1];
>>y = [1, 1; 
    1, 2; 
    1, 3; 
    2, 1;
    2, 2;
    2, 3];
>>row_idx = reshape((x == -1)',1,6);
>>y = y(row_idx,:);

我认为您没有在 y 中包含 x 的所有索引。我将所有这些都包括在 y 中。看看吧。。

通用版：

>> x = [1, 2, 3; -1, 2, -1];
>>y = [1, 1; 
    1, 2; 
    1, 3; 
    2, 1;
    2, 2;
    2, 3];
>>row_idx = reshape((x == -1)',1,size(x,1)*size(x,2));
>>y = y(row_idx,:);