将pandas数据帧转换为列表的字典

Question

我有以下问题.我有一个pandas数据框,它是从pandas.read_sql填充的.

数据框如下所示:

        PERSON              GRADES
20      A                   70
21      A                   23
22      A                   67
23      B                   08
24      B                   06
25      B                   88
26      B                   09
27      C                   40
28      D                   87
29      D                   11

我需要将它转换为这样的列表字典:

{'A':[70,23,67], 'B':[08,06,88,09], 'C':[40], 'D':[87,11]}

我已经尝试了2个小时,现在我没有想法.我想我错过了很简单的事情.

Answer 1

随着groupby和to_dict

df.groupby('PERSON').GRADES.apply(list).to_dict()
Out[286]: {'A': [70, 23, 67], 'B': [8, 6, 88, 9], 'C': [40], 'D': [87, 11]}