Extract a row from a 2D matrix if any of its element matches with elements in other 1D matrix

1 次查看（过去 30 天）
Sachin Shinde
Sachin Shinde 2020-7-3
评论： Sachin Shinde 2020-7-3
I have 2D matrxi A (size: 10000 x 3) and 1D matrix B (size: 5000 x 1)
If one elemnt of row in A matches with any element in B, then that rwo sfould be separated out. Suppose there are N such rows then I want the output as a matrix C (size: N x 3), (which is a subset of A).
I am using the code:
counter = 0
for i = 1:length(A)
if sum(ismember(A(i,:),B)) >= 1
counter = counter + 1;
C(counter,:) = A(i,:);
end
end
This gives me the output I want, but it takes huge time especially on very large sized matrices.
Any one/two liner replacment to speed up the code?
Thanks.

请先登录，再回答此问题。

采纳的回答

Gaurav Aggarwal
Gaurav Aggarwal 2020-7-3
编辑：Gaurav Aggarwal 2020-7-3
Hi Sachin,
Can you check if this works for you?
C = A(any(ismember(A,B),2),:);
Thanks.

更多回答（1 个）

Gifari Zulkarnaen
Try this:
idx = [];
for i = 1:size(A,1)
if any(ismember(A(i,:),B))
idx = [idx i];
end
end
C = A(idx,:);

类别

在 Help Center 和 File Exchange 中查找有关 Numeric Types 的更多信息

标签

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by