熊猫数据框到字典

Question

我有一个像这样的熊猫数据框：

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

需要来自上述数据帧的字典，其中 COL1 将作为键存储，value1 和 value2 将在子字典中。

例如：-

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

Answer 1

您可以使用groupby 和apply 转换为dict ziped 列并最后转换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}}