根据映射文件对数据进行分组的更有效方法

Question

我可以在excel中做到这一点，但我正在寻找一种在python中做到这一点的方法。您知道执行以下操作的方法

初始

          District_1     District_2     District_3
Food      69             47             65
Water     87             86             32
Shelter   63             84             27

映射

District_1      London
District_2      London
District_3      Boston

所需

          London          Boston
Food      116             65
Water     173             32
Shelter   147             27

Answer 1

我将假设您的mapping是一本字典

mapping = {'District_1': 'London', 'District_2': 'London', 'District_3': 'Boston'}

然后将groupby与axis=1一起使用

df.groupby(mapping, axis=1).sum()

         Boston  London
Food         65     116
Water        32     173
Shelter      27     147

将字典传递给groupby时，其字典get的方法将应用于选择的轴（默认为axis = 0），结果将定义组。