python - 如何将熊猫数据框转换为数据字典

Question

我有以下熊猫数据框。我希望转换为默认值或数据字典。

数据框：

    Col A    Col B   Col C
 0   a1     k1       v1  
 1   a1     k2       v2  
 2   a1     k3       v3  
 3   a2     k4       v4  
 4   a3     k5       v5  
 5   a3     k6       v6  

---

预期输出：

# dict
{
    "a1" : {"k1": "v1", k2: "v2", "k3": "v3"},
    "a2" : {"k4": "v4"},
    "a3" : {"k5": "v5", "k6": "v6"}
}

我试过这个没有运气。

dict = df.set_index('Col A').groupby('Col B').apply( lambda x: x.ColC.to_dict()).to_dict()

谢谢您的帮助！！

标签： pythonpandasgroup-by

Answer 1

与groupby_to_dict

{x : y.set_index('ColB')['ColC'].to_dict()for x, y in  df.groupby('ColA')}
Out[252]: 
{'a1': {'k1': 'v1', 'k2': 'v2', 'k3': 'v3'},
 'a2': {'k4': 'v4'},
 'a3': {'k5': 'v5', 'k6': 'v6'}}