将 Dataframe 转换为 Python 中的嵌套字典/树

Question

现在我有一个 CSV 文件，如下所示：

我需要将此数据转换为嵌套字典，其中有不同的级别，具体取决于表中的级别（或列）。 结果应如下所示：

dictionary = {A:{C:1,D:{G:2,H:3},E:4},B:{C:{I:2,J:3},D:10}} 

换句话说，如果该列有一个“子级”，则树中需要有另一个深度级别。

Python中有没有function可以做到这一点？ 或者我需要自己写 function 吗？

Answer 1

如何使用分组

my_dict = df.groupby('A')[['B','C','D']]
       .apply(lambda x: x.set_index('B').to_dict(orient='index'))
       .to_dict()

print (my_dict)