Соответствующая сортировка двух 3D-массивов

Question

Люди. Я не смог найти ответ на мой вопрос.

Допустим, у нас есть два 3D массива:

A=rand(10,10,3); 
B=rand(10,10,3); 

Тогда сортировать по 3 Demension:

[sorted_A, idx] = sort(A,3); 

Как я должен действовать, чтобы установить такой же порядок во втором массиве?

Спасибо!

Answer 1

Вам необходимо линейную индексацию, например:

[ii jj ] = ndgrid(1:size(A,1),1:size(A,2)); 
sorted_B = B(sub2ind(size(A),repmat(ii,[1 1 size(A,3)]),repmat(jj,[1 1 size(A,3)]),idx)); 

Answer 2

С сортировкой по первой матрице вы получаете индекс каждого вида.

Вобще C = B(idx) (Работа для 1D массив)

EDIT

Это для 3D-матрицы, но это решение не векторизации.

C = zeros(10,10,3); 

for dim = 1:3 
    for it = 1:10 
     for jt = 1:10 
      C(it,jt,dim) = B(it,jt,idx(it,jt,dim)); 
     end 
    end 
end