"Desortuj" matrycę. Cofnij sortowanie w Matlab

Question

To pytanie jest w zasadzie rozszerzeniem that question.

mam matrycę A w Matlab i chcą uporządkować tę matrycę wzdłuż jednego wymiaru:

A = rand(3,3,5); [B idx] = sort(A,3);

Teraz idx jest macierzą zawierającą „posortowane” indeksów. Jak mogę odzyskać matrycę A przy użyciu tylko B i ?

The answer pierwotnego pytania nie działa dla macierzy, niestety.

Answer 1

Musisz posortować indeksy idx, aby odzyskać oryginalne indeksy. Reszta prac wymaga uzyskania sformatowanych indeksów wierszy i kolumn odpowiadających wszystkim indeksom dim-3. Realizacja będzie wyglądać mniej więcej tak -

[~,dim3idx] = sort(idx,3); 

[m,n,r] = size(B); 
[rowidx,colidx,~] = ndgrid(1:m,1:n,1:r); 

Aout = B(sub2ind(size(B),rowidx,colidx,dim3idx)) 

Należy pamiętać, że do wykonania, można uzyskać liniowe indeksów generowane przez sub2ind alternatywnie z bsxfun bezpośrednio z parametrami wielkości, a tym samym również uniknąć ndgrid, jak tak -

Aout = B(bsxfun(@plus,bsxfun(@plus,(1:m)',m*(0:n-1)),m*n*(dim3idx-1)))