Ordinare una matrice con un'altra matrice
Domanda
Supponiamo di avere un A
matrice e I ordinare le righe di questa matrice. Come si replica lo stesso ordinamento su un B
matrice (stesse dimensioni ovviamente)? ??
per es.
A = rand(3,4);
[val ind] = sort(A,2);
B = rand(3,4);
%// Reorder the elements of B according to the reordering of A
Questa è la migliore che è venuta in mente
m = size(A,1);
B = B(bsxfun(@plus,(ind-1)*m,(1:m)'));
Per curiosità, alternative?
Aggiornamento: Jonas' soluzione eccellente profilata sul 2008a (XP):
n = n
0.048524 1.4632 1.4791 1.195 1.0662 1.108 1.0082 0.96335 0.93155 0.90532 0.88976
n = 2m
0.63202 1.3029 1.1112 1.0501 0.94703 0.92847 0.90411 0.8849 0.8667 0.92098 0.85569
E 'solo per dimostrare che i cicli non sono un anatema per i programmatori MATLAB Anymore grazie a JITA (forse).
Soluzione
Un modo più chiara per fare questo è quello di utilizzare un ciclo
A = rand(3,4);
B = rand(3,4);
[sortedA,ind] = sort(A,2);
for r = 1:size(A,1)
B(r,:) = B(r,ind(r,:));
end
È interessante notare che la versione ad anello è più veloce per le piccole (<12 righe) e grandi (> ~ 700 righe) matrici quadrate (r2010a, OS X). I più colonne sono relative alle righe, migliore è l'esegue ciclo.
Ecco il codice ho subito hackerato per il test:
siz = 10:100:1010;
tt = zeros(100,2,length(siz));
for s = siz
for k = 1:100
A = rand(s,1*s);
B = rand(s,1*s);
[sortedA,ind] = sort(A,2);
tic;
for r = 1:size(A,1)
B(r,:) = B(r,ind(r,:));
end,tt(k,1,s==siz) = toc;
tic;
m = size(A,1);
B = B(bsxfun(@plus,(ind-1)*m,(1:m).'));
tt(k,2,s==siz) = toc;
end
end
m = squeeze(mean(tt,1));
m(1,:)./m(2,:)
Per le matrici quadrate
ans =
0.7149 2.1508 1.2203 1.4684 1.2339 1.1855 1.0212 1.0201 0.8770 0.8584 0.8405
Per doppio delle colonne quante sono le righe (stesso numero di righe)
ans =
0.8431 1.2874 1.3550 1.1311 0.9979 0.9921 0.8263 0.7697 0.6856 0.7004 0.7314
Altri suggerimenti
Sort () restituisce l'indice lungo la dimensione che si Ordinati. È possibile costruire esplicitamente indici per le altre dimensioni che causano le righe rimangono stabili, e quindi utilizzano indicizzazione lineare per riorganizzare l'intera matrice.
A = rand(3,4);
B = A; %// Start with same values so we can programmatically check result
[A2 ix2] = sort(A,2);
%// ix2 is the index along dimension 2, and we want dimension 1 to remain unchanged
ix1 = repmat([1:size(A,1)]', [1 size(A,2)]); %//'
%// Convert to linear index equivalent of the reordering of the sort() call
ix = sub2ind(size(A), ix1, ix2)
%// And apply it
B2 = B(ix)
ok = isequal(A2, B2) %// confirm reordering
Non puoi semplicemente fare questo?
[val ind]=sort(A);
B=B(ind);
Ha funzionato per me, a meno che non sto capendo il problema sbagliato.