группы по словарю в панде 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

Быстрое решение заключается в сохранении в новой переменной в результате DataFrame (grouped_data в следующем коде), а затем присвоить Name колонки к нему с помощью оригинал DataFrame:

grouped_data = mb2.groupby(by=mapping,axis=1).sum() 
grouped_data['Name'] = mb2['Name'] 
print(grouped_data) 

EDIT: Другой способ достижения того же результата, чтобы изменить mapping словарь, добавив "Name": "Name" пару к нему, и призывая mb2.groupby(by=mapping,axis=1).sum() как раньше.

Answer 2

Вы можете использовать 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