matlab - "Desort"一个矩阵。在 Matlab 中撤消排序

Question

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

我在 Matlab 中有一个矩阵 A 并且想沿一维对该矩阵进行排序:

A = rand(3,3,5);
[B idx] = 排序(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)))