How to obtain total sum by group

Hi, Could someone explain how to compute the total sum by group (without using for loop)?
A example is: groupid = [1; 1; 1; 2; 2; 3; 3;3]; value=[1;2;3;4;5;6;7;8]. the desired result is: runsum = [6;6;6;9;9; 21;21;21].
I am trying to get the result without using loop. Can "accumarray" do this?
Thanks a lot
Siying

 Accepted Answer

Siying - yes, accumarray can be used to sum elements of the same group. For example, we can do
groupSums = accumarray(groupid,value);
which gives us
groupSums =
6
9
21
with your runsum as
runsum = groupSums(groupid);

3 Comments

Thank you very much! I did know how to obtain groupSums, but I didn't know how to expand that according to the groupid. The last line of the code is very clean and helpful!
I have one more following up question. The last line seems to work only if the groupSums is a vector, is there a way I can expand this method to a matrix? Thank you in advance!
Which is the matrix: the group ids or values, or both? Please provide an example.

Sign in to comment.

More Answers (2)

Yes. This should work; it's probably not the shortest way to do it, but it avoids looping.
key = unique(groupid);
tsum = accumarray(groupid,value);
[~,idx] = ismember(groupid,key);
runsum = tsum(idx);
group_sum = accumarray(groupid, value);
runsum = group_sum(A);

1 Comment

Yes! I did read that thread! The problem I have is don't know how to covert the group sum to the matrix I need without using a loop.
Thanks!

Sign in to comment.

Categories

Find more on Loops and Conditional Statements in Help Center and File Exchange

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!