python - 按 Pandas DataFrame 中的字典分组

Question

我从一个excel文件中读取数据，是这样的:

我想得到每个员工的 Q1 和 Q2 的总和，所以我创建了一个字典，并想根据它进行分组，

import pandas as pd

mb2 = pd.read_excel('C:\\Users\\IBM_ADMIN\\Desktop\\ml-1m\\工作簿1.xlsx', sheetname='Sheet3')
mapping = {'Jan':'Q1','Feb':'Q1','Mar':'Q1','Apr':'Q2','May':'Q2','Jun':'Q2'}
mb2.groupby(by=mapping,axis=1).sum()

如何同时显示字段名称？我尝试这样做，但没有成功:

mb2.groupby(by=(mapping,'Name'),axis=1).sum()

Answer 1

您可以使用 set_index首先，然后是 groupby，最后是 reset_index :

df = mb2.set_index('Name').groupby(by=mapping,axis=1).sum().reset_index()
print (df)
      Name    Q1    Q2
0     Tony  1367  1835
1    Rafal  1405  2574
2     John  1550  1409
3  Katrina  1555  1509
4    Kathy  1583  1491
5      Tom  1344  1874