Pandas dataframe group by multiple columns

Viewed 10220

Given a dataframe with two datetime columns A and B and a numeric column C, how to group by month of both A and B and sum(C) i.e.

In [1]: df
Out[1]: 

      A           B            C  
0  2013-01-01  2013-01-01  0.282863 
1  2013-01-02  2013-01-01  0.173215 
2  2013-02-03  2013-02-04  2.104569 
3  2013-02-09  2013-04-15  0.706771 
4  2013-03-05  2013-08-01  0.567020 
5  2013-03-06  2013-04-01  0.113648
4 Answers

By using groupby

df.groupby([df.A.dt.month,df.B.dt.month]).C.sum()

Out[954]: 
A  B
1  1    0.456078
2  2    2.104569
   4    0.706771
3  4    0.113648
   8    0.567020
Name: C, dtype: float64

Note: By using this , make sure A and B are datetime format If not , do following code before groupby

df.A=pd.to_datetime(df.A)
df.B=pd.to_datetime(df.B) 
df['month_A'] = [i.month for i in pd.to_datetime(df.A)]
df['month_B'] = [i.month for i in pd.to_datetime(df.B)]

df.groupby(['month_A', 'month_B']).sum()

If you combine with following, you will get back the result with the respective values in A and B column

idsum = df.groupby([df.A.dt.month,df.B.dt.month])["C"].transform(sum) == df["C"]
df=df[idsum]
Related