MATLAB：按另一個 3d 數組對 3d 數組進行排序

Question

我有一個隨機數的 3d 數組 A，我想單獨訂購每個第 k 個維度：

A=rand(3,1,16);
[m, n, k]=size(A);

指示每個矩陣在第 3 維中的順序的數組是 B：

B=randi(3,3,1,16); %this should be without replacement but think it will work anyway

如果 A(:,1,1)=[0.5, 0.2, 0.6]' 和 B(:,1,1)=[3,1,2] 那么有序 A 應該是 [0.2, 0.6, 0.5]' 和對於每個 A(:,1,1:k)，依此類推。 請注意，它不是以數字方式訂購 A。

A（B）是我可能期望的工作並保持尺寸但不是正確的順序。

我試圖解決這個問題： https://uk.mathworks.com/matlabcentral/answers/307838-sort-3d-matrix-according-to-another-3d-matrix沒有任何成功。

任何想法將不勝感激。

Answer 1

根據B沿第k維對A進行排序：如果您的數據不復雜，一種簡單的方法是：

1. A和B打包成一個復數數組，其中A是虛部， B是實部；
2. 根據實部沿第k維排序；
3. 保留虛部。

這是在一行中完成的，如下所示：

result = imag(sort(B+1j*A, k, 'ComparisonMethod', 'real'));

示例：

>> A = rand(2,4,3)
A(:,:,1) =
   0.162182308193243   0.311215042044805   0.165648729499781   0.262971284540144
   0.794284540683907   0.528533135506213   0.601981941401637   0.654079098476782
A(:,:,2) =
   0.689214503140008   0.450541598502498   0.228976968716819   0.152378018969223
   0.748151592823709   0.083821377996933   0.913337361501670   0.825816977489547
A(:,:,3) =
   0.538342435260057   0.078175528753184   0.106652770180584   0.004634224134067
   0.996134716626885   0.442678269775446   0.961898080855054   0.774910464711502

>> B = randi(9, size(A))
B(:,:,1) =
     8     1     3     4
     8     4     8     9
B(:,:,2) =
     2     2     8     5
     3     2     6     2
B(:,:,3) =
     8     4     4     3
     6     5     1     2

>> k = 2;

>> result = imag(sort(B+1j*A, k, 'ComparisonMethod', 'real'))
result(:,:,1) =
   0.311215042044805   0.165648729499781   0.262971284540144   0.162182308193243
   0.528533135506213   0.601981941401637   0.794284540683907   0.654079098476782
result(:,:,2) =
   0.450541598502498   0.689214503140008   0.152378018969223   0.228976968716819
   0.083821377996933   0.825816977489547   0.748151592823709   0.913337361501670
result(:,:,3) =
   0.004634224134067   0.078175528753184   0.106652770180584   0.538342435260057
   0.961898080855054   0.774910464711502   0.442678269775446   0.996134716626885