python – 熊猫数据帧到dict的dict

鉴于以下pandas数据框：

  ColA ColB  ColC
0   a1    t     1
1   a2    t     2
2   a3    d     3
3   a4    d     4

我想要一本字典词典.

但我设法创建了以下内容：

d = {t : [1, 2], d : [3, 4]}

通过：

d = {k: list(v) for k,v in duplicated.groupby("ColB")["ColC"]}

我怎么能得到字典的字典：

dd = {t : {a1:1, a2:2}, d : {a3:3, a4:4}}

解决方法:

您可以事先使用groupby应用步骤执行此操作.

dd = df.set_index('ColA').groupby('ColB').apply(
    lambda x: x.ColC.to_dict()
).to_dict()

或者,通过词典理解：

dd = {k : g.ColC.to_dict() for k, g in df.set_index('ColA').groupby('ColB')}

print(dd)
{'d': {'a3': 3, 'a4': 4}, 't': {'a1': 1, 'a2': 2}}