“对”矩阵进行“分解”。在 Matlab 中撤消排序

Question

这个问题基本上是that question的延伸。

我在 Matlab 中有一个矩阵 A，我想对该矩阵进行一维排序：

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

现在idx 是一个包含“排序”索引的矩阵。如何仅使用 B 和 idx 取回矩阵 A？

不幸的是，原始问题的The answer 不适用于矩阵。

Answer 1

您需要对索引idx进行排序以取回原始索引。其余的工作将涉及获取与所有这些 dim-3 索引相对应的格式化行和列索引。实现看起来像这样 -

[~,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))

请注意，出于性能考虑，可以直接从大小参数中获取sub2ind 或bsxfun 生成的线性索引，因此也可以避免ndgrid，就像这样 -

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