Azzera filtri
Azzera filtri

Eliminating a FOR loop

2 visualizzazioni (ultimi 30 giorni)
Scott
Scott il 2 Set 2014
Modificato: Andrei Bobrov il 2 Set 2014
I am trying to eliminate a FOR loop to improve speed. The problem I am working with involves evaluating each element of an array with each element of a matrix. Below is a simplified example of what I am trying to do.
I have two matrices:
A = [a11 a12 a13;a21 a22 a23; a31 a32 a33];
B = [b1 b2 b3 b4];
What I would like to do is evaluate each element of B against each element of A. For example
for i=1:1:4
[I,J] = find(B(i)==A(:,:));
end
Is there a way to do this without a FOR loop.
Thanks in advance,
Scott
  1 Commento
Guillaume
Guillaume il 2 Set 2014
Modificato: Guillaume il 2 Set 2014
Your example overwrites I and J on each pass of the loop, so it's not clear what you want out. Did you mean:
[I{i} J{i}] = find(B(i)==A);

Accedi per commentare.

Risposta accettata

Matt J
Matt J il 2 Set 2014
Modificato: Matt J il 2 Set 2014
L=bsxfun(@eq,A(:),B(:).');
L=reshape(L,[size(A),length(B)]);
[I,J,~]=ind2sub(size(L),find(L));
  1 Commento
Scott
Scott il 2 Set 2014
Matt,
Thank you very much. Your answer was exactly what I was looking for.

Accedi per commentare.

Più risposte (4)

Iain
Iain il 2 Set 2014
The function you might be looking for is probably
intersect

Pierre Benoit
Pierre Benoit il 2 Set 2014
Modificato: Pierre Benoit il 2 Set 2014
[~,idx] = ismember(B,A);
[I,J] = ind2sub(size(A),idx);

Guillaume
Guillaume il 2 Set 2014
You can use arrayfun or bsxfun to replace a loop. It's not necessary faster but may make the intent of the code clearer. In this case, assuming you want I and J as cell arrays, not so:
[I, J] = arrayfun(@(b) ind2sub(size(A), find(b==A)), B, 'uni', false);

Andrei Bobrov
Andrei Bobrov il 2 Set 2014
Modificato: Andrei Bobrov il 2 Set 2014
A = randi(10,5,6);
B = randi(10,4,1);
[i1,j1] = find(bsxfun(@eq,A,reshape(B,1,1,[])));
out_ij = [i1, rem(j1 - 1,size(A,2)) + 1];

Categorie

Scopri di più su Loops and Conditional Statements in Help Center e File Exchange

Community Treasure Hunt

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

Start Hunting!

Translated by