通过将多个列求和在一起来实现分组

Question

我有一套伦敦犯罪数据需要分析。下面的dataframe是dataframe的一个片段。我试图找出2018年4个月内每个地区所有犯罪的总和

Crime_type                | Crime sub_type   | District   | 201803 | 201804 | 201805 | 201806 |
----------------------------------------------------------------------------------------------
Arson and Criminal Damage |      Arson      | Westminster |   45   |   90   |    54  |   36   |
Arson and Criminal Damage | Criminal Damage | Westminster |   64   |   35   |    74  |   63   |
Arson and Criminal Damage |      Arson      |   Bexley    |   54   |   19   |    23  |   98   |
Arson and Criminal Damage | Criminal Damage |   Bexley    |   98   |   63   |    89  |   87   |    

所需的输出将是：

Westminster: 461
Bexley: 531

我知道Pandas group by函数可以完成我试图实现的功能，但我不确定如何在本例中正确使用它，找到多个列的总和。我将其按地区分组，并在每个4个月栏中查找总和，但我希望在所有4个栏中查找总和，而不是单独查找

这是我用来实现上述功能的代码

df.groupby(['District'])[["201803", "201804", "201805", "201806"]].sum()

Answer 1

一个解决方案（类似于评论）是：

year_months_cols = ["201803", "201804", "201805", "201806"]
df["sum_months"] = df[year_months_cols].sum(axis=1)
df.groupby("District")["sum_months"].agg("sum")