python之Pandas Dataframe 按列分组

我有一个这样的数据框:

Subject_id    Subject    Score     
Subject_1        Math        5                  
Subject_1    Language        4                  
Subject_1       Music        8 
Subject_2        Math        8                  
Subject_2    Language        3                  
Subject_2       Music        9 

我想把它转换成字典，按 subject_id 分组

{'Subject_1': {'Math': 5, 
               'Language': 4, 
               'Music': 8}, 
{'Subject_2': {'Math': 8, 
               'Language': 3, 
               'Music': 9} 
} 

如果我只有一个主题，那么我可以:

my_dict['Subject_1'] = dict(zip(df['Subject'],df['Score'])) 

但是因为我有几个主题，键列表重复，所以我不能直接使用 zip。

Dataframes 有一个 .to_dict('index') 方法，但我需要能够在创建字典时按特定列进行分组。

我怎样才能做到这一点？

谢谢。

请您参考如下方法:

使用groupby使用自定义 lambda 函数和最后转换输出 Series to_dict :

d = (df.groupby('Subject_id') 
       .apply(lambda x: dict(zip(x['Subject'],x['Score']))) 
       .to_dict()) 
 
print (d) 
{'Subject_2': {'Math': 8, 'Music': 9, 'Language': 3},  
 'Subject_1': {'Math': 5, 'Music': 8, 'Language': 4}} 

详细信息:

print (df.groupby('Subject_id').apply(lambda x: dict(zip(x['Subject'],x['Score'])))) 
 
Subject_id 
Subject_1    {'Math': 5, 'Music': 8, 'Language': 4} 
Subject_2    {'Math': 8, 'Music': 9, 'Language': 3} 
dtype: object