python - Pandas 数据框到字典

Question

我有一个 pandas 数据框，例如:

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 zip 列并最后转换 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}}