將 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)