python-3.x - 将 pandas 数据框转换为列表字典

Question

我遇到以下问题。我有一个从 pandas.read_sql 填充的 pandas 数据框。

数据框如下所示:

        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]}