python - Pandas 数据框到字典，值作为列表

Question

我们有以下数据框 df，

   0   1  2  3
0 'a' 'b' 1  2
1 'a' 'c' 2  4

我想要输出为

{('a', 'b'): [1, 2], ('a', 'c'): [2, 4]}

如何使用 .to_dict() 生成输出？

我试过了

df.set_index([0,1])[[2,3]].to_dict()

并得到结果

{2: {('a', 'b'): 1, ('a', 'c'): 2}, 3: {('a', 'b'): 2, ('a', 'c'): 4}}

Answer 1

设置索引后，进行转置，然后使用to_dict和orient='list':

df.set_index([0,1]).T.to_dict(orient='list')

结果输出:

{('a', 'b'): [1, 2], ('a', 'c'): [2, 4]}