Question

给出以下向量：

5 4 1 2 3 1 4 5 3 2 3 2 1 5 4
_________ _________ _________

我想对向量的每5个元素应用排序。因此输出将是：

1 2 3 4 5 1 2 3 4 5 1 2 3 4 5 
_________ _________ _________

如何在MATLAB中实现无循环？

P.S。我还想提取排序索引，以将其应用于另一个向量。

Answer 1

如果要避免循环，可以结合使用reshape和sort来实现所需的目标：

b = [5 4 1 2 3 1 4 5 3 2 3 2 1 5 4];
b2 = reshape(b, [5 3]);           % Reshape your array to be a [5x3] matrix
b2_new = sort(b2, 1);             % Sort each column of your matrix seperately
b_new = reshape(b2_new, size(b)); % Reshape the outcome back to the original dimensions

或者全部一行：

b_new = reshape(sort(reshape(b, [5 3]), 1), size(b));

当然，您必须更改数字5和3以适合您的问题。重要的是要确保您为重塑命令输入的第一个值（在这种情况下为5）等于您要排序的子向量的长度，因为Matlab是列主行。

编辑：

如果要排序一个特定的向量，然后对其他向量应用相同的重新排序，则可以使用sort函数的可选第二个输出参数。使用与上述相同的向量：

b = [5 4 1 2 3 1 4 5 3 2 3 2 1 5 4];
b2 = reshape(b, [5 3]);

产量：

假设您要对第一列进行排序，并对第二和第三列应用相同的重新排序，那么您可以这样做：

[~,idx] = sort( b2(:,1) );   % Sorts the first column of b2, and stores the index map in 'idx'

这将产生idx = [3 4 5 2 1]。现在，您可以使用这些索引对所有列进行排序：

b2_new = b2(idx,:);
b2_new =
     1     5     1
     2     3     5
     3     2     4
     4     4     2
     5     1     3

最后，您可以使用reshape回到原始尺寸：

b_new = reshape(b2_new, size(b));

编辑2：

如果您想整体存储b的重新排序，并将其应用于新的向量c，我们将不得不变得更有创意。以下是一种方法：

b = [5 4 1 2 3 1 4 5 3 2 3 2 1 5 4];
b2, = reshape(b, [5 3]);

% Sort each column of your matrix seperately, and store the index map
[~,idx] = sort(b2, 1);

% Alter the index map, such that the indices are now linear indices:
idx = idx + (0:size(idx,2)-1)*size(idx,1);

% Reshape the index map to the original dimensions of b:
idx = reshape(idx, size(b));

% Now sort any array you want using this index map as follows:
b_new = b(idx);
c_new = c(idx);

对向量的每个范围进行排序

1 个答案: