Quick way to sort an array with respect to row sum in Matlab

淺唱寂寞╮ 提交于 2019-12-13 18:11:48

问题


Here's a small example of what I want. Given the following array:

1 1 2
2 2 1
1 1 1
1 1 6

Sorted (row sum is shown in parenthesis):

1 1 6 (8)
2 2 1 (5)
1 1 2 (4)
1 1 1 (3)

Is there a quick way to achieve this in Matlab?


回答1:


Since sort returns the indexes in order as well as the sorted matrix, you can use these indices to access the original data -- try this:

% some data
A = [
  1 1 2;
  2 2 1;
  1 1 1;
  1 1 6;
];

% compute the row totals
row_totals = sum(A,2);

% sort the row totals (descending order)
[sorted, row_ids] = sort(row_totals, 'descend');

% and display the original data in that order (concatenated with the sums)
disp([A(row_ids,:), row_totals(row_ids)])

>>> 
 1     1     6     8
 2     2     1     5
 1     1     2     4
 1     1     1     3



回答2:


The ugliest one-liner I could come up with:

>> subsref( sortrows( [sum(A,2) A], -1 ), struct('type','()','subs',{{':',1+(1:size(A,2))}}) )

ans =

 1     1     6
 2     2     1
 1     1     2
 1     1     1

Disclaimer: I don't think anyone should write this kind of code, but it's a nice practice to keep your Matlab's skills sharp.




回答3:


Just do something very simple like follows

temp = [1 1 2
        2 2 1
        1 1 1
        1 1 6];
rowSums = sum(temp,2);
[~,idx] = sort(rowSums,'descend');
output = [temp(idx,:),rowSums(idx)];

EDIT

Changed the above code to make sure the sum is appended to the last column. I did not notice that this was a requirement when I read the question initially.




回答4:


I leave it for you to judge if this is uglier than @Shai's:

fliplr(diff([sortrows(fliplr(-cumsum(A,2))) zeros(size(A,1),1) ],1,2))



回答5:


Let's do some matrix multiplication

>> sortrows([sum(A,2) A], -1)*[zeros(1,size(A,2)); eye(size(A,2))]

returns

ans =
     1     1     6
     2     2     1
     1     1     2
     1     1     1


来源:https://stackoverflow.com/questions/18721094/quick-way-to-sort-an-array-with-respect-to-row-sum-in-matlab

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!