Pandas groupby cumulative sum

Question

I would like to add a cumulative sum column to my Pandas dataframe so that:

name | day       | no

-----|-----------|----

Jack | Monday    | 10

Jack | Tuesday   | 20

Jack | Tuesday   | 10

Jack | Wednesday | 50

Jill | Monday    | 40

Jill | Wednesday | 110

becomes:

Jack | Monday     | 10  | 10

Jack | Tuesday    | 30  | 40

Jack | Wednesday  | 50  | 90

Jill | Monday     | 40  | 40

Jill | Wednesday  | 110 | 150

I tried various combos of df.groupby and df.agg(lambda x: cumsum(x)) to no avail.

Answer 1

Use groupby() twice:

In [52]:

print df

   name        day   no

0  Jack     Monday   10

1  Jack    Tuesday   20

2  Jack    Tuesday   10

3  Jack  Wednesday   50

4  Jill     Monday   40

5  Jill  Wednesday  110

In [53]:

print df.groupby(by=['name','day']).sum().groupby(level=[0]).cumsum()

                 no

name day           

Jack Monday      10

     Tuesday     40

     Wednesday   90

Jill Monday      40

     Wednesday  150