Eliminating a FOR loop

5 Ansichten (letzte 30 Tage)
Scott
Scott am 2 Sep. 2014
Bearbeitet: Andrei Bobrov am 2 Sep. 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 Kommentar
Guillaume
Guillaume am 2 Sep. 2014
Bearbeitet: Guillaume am 2 Sep. 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);

Melden Sie sich an, um zu kommentieren.

Akzeptierte Antwort

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

Melden Sie sich an, um zu kommentieren.

Weitere Antworten (4)

Iain
Iain am 2 Sep. 2014
The function you might be looking for is probably
intersect

Pierre Benoit
Pierre Benoit am 2 Sep. 2014
Bearbeitet: Pierre Benoit am 2 Sep. 2014
[~,idx] = ismember(B,A);
[I,J] = ind2sub(size(A),idx);

Guillaume
Guillaume am 2 Sep. 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 am 2 Sep. 2014
Bearbeitet: Andrei Bobrov am 2 Sep. 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];

Kategorien

Mehr zu Loops and Conditional Statements finden Sie in Help Center und File Exchange

Community Treasure Hunt

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

Start Hunting!

Translated by