熊猫:按两列分组以获得另一列的总和

add*_*ons 6 group-by pandas

我查看了之前提出的大多数问题,但无法找到我的问题的答案:

我有以下data.frame

           id   year month score num_attempts
0      483625  2010    01   50      1
1      967799  2009    03   50      1
2      213473  2005    09  100      1
3      498110  2010    12   60      1
5      187243  2010    01  100      1
6      508311  2005    10   15      1
7      486688  2005    10   50      1
8      212550  2005    10  500      1
10     136701  2005    09   25      1
11     471651  2010    01   50      1
Run Code Online (Sandbox Code Playgroud)

我想获得以下数据框

year month sum_score sum_num_attempts
2009    03   50           1
2005    09  125           2
2010    12   60           1
2010    01  200           2
2005    10  565           3
Run Code Online (Sandbox Code Playgroud)

这是我尝试过的:

sum_df = df.groupby(by=['year','month'])['score'].sum()
Run Code Online (Sandbox Code Playgroud)

但这看起来并不高效且正确。如果我需要汇总多个列,这似乎是一个非常昂贵的电话。例如,如果我有另一列,num_attempts并且只想按年份月份求和。

Den*_*zov 6

这应该是一种有效的方法:

sum_df = df.groupby(['year','month']).agg({'score': 'sum', 'num_attempts': 'sum'})
Run Code Online (Sandbox Code Playgroud)

  • 如果有人还想重命名创建的列,以便新的(分数)列不采用与输入列相同的名称。在末尾附加 `.rename(columns={'score': 'score_sum_by_year_month'})` 。 (2认同)