Эффективно использовать python groupby или defaultdict?

Question

У меня есть csv с именем, ролью, многолетним опытом. Я хочу создать список кортежей, который объединяет (name, role1, total_exp_inthisRole) для всех работодателей. до сих пор я могу использовать defaultdict сделать ниже

import csv, urllib2 
from collections import defaultdict 

response = urllib2.urlopen(url) 
cr = csv.reader(response) 
parsed = ((row[0],row[1],int(row[2])) for row in cr)  
employees =[] 
for item in parsed: 
    employees.append(tuple(item)) 
employeeExp = defaultdict(int) 
for x,y,z in employees: # variable unpacking 
    employeeExp[x] += z  
employeeExp.items() 

выхода: [('Ken', 15), ('Buckky', 5), ('Tina', 10)]

но как я использую второй столбец также для достижения результата я хочу. Могу ли я попытаться решить groupby возможно несколько ключей или более простой способ? Спасибо всем заранее.

Answer 1

Вы можете просто передать кортеж имя и роль в вашей defaultdict, а не только один пункт:

for x,y,z in employees: 
    employeeExp[(x, y)] += z 

Для вашего второго ожидаемого выхода ([('Ken', ('engineer', 5),('sr. engineer', 6)), ...])

Вам необходимо агрегировать результат вышеупомянутый отрывок еще один раз, но на этот раз вам нужно использовать defaultdict со списком:

d = defaultdict(list) 

for (name, rol), total_exp_inthisRole in employeeExp.items(): 
    d[name].append(rol, total_exp_inthisRole)