2012-05-09 9 views
11

Załóżmy, że mam macierz (m x n) Q i wektor wierszowy r, np.znajdowanie pasujących wierszy w macierzy

Q = [ 1 2 3 ; 4 2 3 ; 5 6 7 ; 1 2 3 ; 1 2 3 ; 1 2 5 ]; 

r = [ 1 2 3 ]; 

co jest najprostszym sposobem uzyskania logiczny wektor (o długości m), które wskazuje, który z wierszy Q są identyczne (dla wszystkich elementów) do określonego rzędu R?

W przypadku próbki powyżej, który powinien być

[ 1 0 0 1 1 0 ]; 
+0

Bardzo blisko związany z: http://stackoverflow.com/questions/6209904/find-given-row-in-a-matrix – neuronet

Odpowiedz

19

Można użyć ismember i zrobić w jednej linii:

>> ismember(Q,r,'rows')' 

ans = 

    1  0  0  1  1  0 
13
all(bsxfun(@eq, r, Q),2)' 

bsxfun(@eq, r, Q) porównuje każdy wiersz i zwraca matrycę z tej samej wielkości, co Q:

>> bsxfun(@eq, r, Q) 

ans = 

    1  1  1 
    0  1  1 
    0  0  0 
    1  1  1 
    1  1  1 
    1  1  0 

funkcja all oblicza czy wynik bsxfun jest prawdziwy w każdym rzędzie osobno. W ten sposób zwraca:

>> all(ans,2)' 

ans = 

    1  0  0  1  1  0 

i tak, istnieje również operator transpozycji ' aby dopasować pożądany wynik rzędzie

+1

Warto zauważyć, że jest to również nieco szybsze niż ismember –

0

łatwiejszy sposób z repmat:

a = [1 2 3; 4 5 6; 7 8 9]; 
t = [4 5 6]; 
[x,y] = size(a); 
r = all(a==repmat(t,y,1), 2)' 
+0

również mniej wydajne i wolniejsze, zobacz: http://blogs.mathworks.com/loren/2008/08/04/comparing-repmat-and-bsxfun-performance/#9 –

+0

Skuteczność tam, gdzie nie jest potrzebna, jest przekleństwem. Prostota pisania i rozumowania odnosi się bezpośrednio do lepszego kodu. – Castilho

+0

co jest tak trudno zrozumieć o 'bsxfun (@eq, r, Q)' ?? Jeśli nauczysz się go na takich łatwych przykładach, skorzystasz z niego później, gdy będziesz aplikować na złożone problemy. –

1
a = [1 1 1; 2 2 2; 3 3 3]; 
b = a(1:2,;); 
[temp locb] = ismember(a,b,'rows'); 
b(locb(locb~=0),:) 

ans = 

    1  1  1 
    2  2  2 
Powiązane problemy