Cumulative sum within group, reset when encounters a 0

Question

Namrata Goswami 2020-12-17

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/696195-cumulative-sum-within-group-reset-when-encounters-a-0

回答： Eric Sofen 2021-1-6

I have a table and I want to get the cumulative sum within a group(by ID), but the cumulative count should reset if the counter is 0 at any point within a group and again start the cumulative count from 1.

Input

ID Counter

A 1

A 0

A 1

B 1

B 0

B 1

Expected output:

ID Counter Cumulative

A 1 1

A 0 0

A 1 1

A 1 2

B 1 1

B 0 0

B 1 1

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Eric Sofen 2021-1-6

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/696195-cumulative-sum-within-group-reset-when-encounters-a-0#answer_592218

在 MATLAB Online 中打开

It took me a while to come up with a way to do this without looping, then it clicked! You need to create groups based on where the 0s are. To do that, flip the logic of the counter and cumsum that:

t.ResetGroup = cumsum(~t.Counter);

Then, you can use varfun to group by both ID and ResetGroup and do the cumsum.

t = varfun(@cumsum, t, "GroupingVariable",["ID","ResetGroup"])