Clear Filters
Clear Filters

How to obtain total sum by group

35 views (last 30 days)
Siying Liu
Siying Liu on 18 Dec 2015
Commented: Geoff Hayes on 22 Dec 2015
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

Geoff Hayes
Geoff Hayes on 18 Dec 2015
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
Siying Liu
Siying Liu on 22 Dec 2015
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!
Geoff Hayes
Geoff Hayes on 22 Dec 2015
Which is the matrix: the group ids or values, or both? Please provide an example.

Sign in to comment.

More Answers (2)

jgg
jgg on 18 Dec 2015
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);

Walter Roberson
Walter Roberson on 18 Dec 2015
group_sum = accumarray(groupid, value);
runsum = group_sum(A);
  1 Comment
Siying Liu
Siying Liu on 18 Dec 2015
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 Entering Commands 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!