Как преобразовать pandas dataframe в datadict

Question

У меня есть следующий кадр данных панд. Я хочу преобразовать в значение по умолчанию или datadict.

Дата кадра:

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

Спасибо за помощь !!

Answer 1

Используйте groupby и перебирайте группы:

{k: dict(g[['Col B', 'Col C']].values) for k, g in df.groupby('Col A')}

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

Answer 2

Использование 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'}}