熊猫数据框到字典

Posted 2023-03-11

【中文标题】熊猫数据框到字典

【英文标题】：Pandas dataframe to dictionary

【发布时间】：2016-10-17 13:43:51

【问题描述】：

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

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

【参考方案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