熊貓據幀到字典

Question

我有一個熊貓數據幀，如：

COL1 VALUE1 VALUE2 
A  A12  1  
B  B13  2 
A  C12  3 
B  Q12  4 

需要一個詞典從上面的數據幀，其中COL1將被存儲爲鍵和值1和值2將在子字典。

如： -

dict = {'A':{'A12':1, 'C12':3}, B:{'B13':2, 'Q12':4}} 

Answer 1

您可以使用groupby與apply轉化爲dictzip編列和最後轉換to_dict：

d = df.groupby('COL1').apply(lambda x: dict(zip(x.VALUE1, x.VALUE2))).to_dict() 
print (d) 
{'A': {'C12': 3, 'A12': 1}, 'B': {'B13': 2, 'Q12': 4}}