用另一个矩阵对一个矩阵进行排序答案

【问题标题】：Sort a matrix with another matrix用另一个矩阵对一个矩阵进行排序
【发布时间】：2011-02-10 08:43:40
【问题描述】：

假设我有一个矩阵A，我对该矩阵的行进行排序。如何在矩阵B（当然大小相同）上复制相同的排序？

例如

A = rand(3,4);
[val ind] = sort(A,2);
B = rand(3,4);
%// Reorder the elements of B according to the reordering of A

这是我想出的最好的

m = size(A,1);
B = B(bsxfun(@plus,(ind-1)*m,(1:m)'));

出于好奇，还有其他选择吗？

更新： Jonas' excellent solution 在 2008a (XP) 上配置：

n = n

0.048524       1.4632       1.4791        1.195       1.0662        1.108       1.0082      0.96335      0.93155      0.90532      0.88976

n = 2m

0.63202       1.3029       1.1112       1.0501      0.94703      0.92847      0.90411       0.8849       0.8667      0.92098      0.85569

感谢JITA（也许），这只是表明循环不再是 MATLAB 程序员的诅咒。

【问题讨论】：

大声笑 .. 我花了一秒钟才意识到您指的是另一种语言 .. 想知道我在看什么深奥的 MATLAB 语法！

标签： matlab bsxfun

【解决方案1】：

一个更清晰的方法是使用循环

A = rand(3,4);
B = rand(3,4);
[sortedA,ind] = sort(A,2);

for r = 1:size(A,1)
   B(r,:) = B(r,ind(r,:));
end

有趣的是，对于小型（~700 行）方形阵列（r2010a，OS X），循环版本更快。相对于行的列越多，循环执行的越好。

这是我快速编写的用于测试的代码：

siz = 10:100:1010;
tt = zeros(100,2,length(siz));

for s = siz
    for k = 1:100

        A = rand(s,1*s);
        B = rand(s,1*s);
        [sortedA,ind] = sort(A,2);

        tic;
        for r = 1:size(A,1)
            B(r,:) = B(r,ind(r,:));
        end,tt(k,1,s==siz) = toc;

        tic;
        m = size(A,1);
        B = B(bsxfun(@plus,(ind-1)*m,(1:m).'));
        tt(k,2,s==siz) = toc;

    end
end

m = squeeze(mean(tt,1));

m(1,:)./m(2,:)

对于方阵

ans =

    0.7149    2.1508    1.2203    1.4684    1.2339    1.1855    1.0212    1.0201    0.8770       0.8584    0.8405

列数是行数的两倍（行数相同）

ans =

    0.8431    1.2874    1.3550    1.1311    0.9979    0.9921    0.8263    0.7697    0.6856    0.7004    0.7314

【讨论】：

+1 - 令人印象深刻！我想 JIT 现在非常强大。我很快就会发布 R2008a 的结果。

【解决方案2】：

Sort() 沿您排序的维度返回索引。您可以为使行保持稳定的其他维度显式构建索引，然后使用线性索引重新排列整个数组。

A = rand(3,4);
B = A; %// Start with same values so we can programmatically check result

[A2 ix2] = sort(A,2);
%// ix2 is the index along dimension 2, and we want dimension 1 to remain unchanged
ix1 = repmat([1:size(A,1)]', [1 size(A,2)]); %//'
%// Convert to linear index equivalent of the reordering of the sort() call
ix = sub2ind(size(A), ix1, ix2) 
%// And apply it
B2 = B(ix)
ok = isequal(A2, B2) %// confirm reordering

【讨论】：

不错。比较这个解决方案的时间也会很有趣。
从表面上看，它看起来和我的非常相似，除了我手动计算线性索引并使用bsxfun。
遗憾的是，它在性能方面是一个失败者。在我的 R2010a 上，这比循环或 bsxfun 解决方案慢 25%。
@Jacob：我认为你是对的，我只是在阅读 bsxfun() 代码时遇到了麻烦。毕竟这似乎不是对 OP 的改进。
@Jacob: repmat 只有一个功能。当我读到repmat 时，我知道会发生什么。当我阅读bsxfun 时，我知道我现在必须真正注意。话虽如此，每次写信bsxfun 都让我很开心。 bsxfun bsxfun bsxfun

【解决方案3】：

你不能这样做吗？

[val ind]=sort(A);
B=B(ind);

它对我有用，除非我理解你的问题是错误的。

【讨论】：

ind 仅包含行索引。 B = B(ind) 没有给出所需的结果。