Pandas DataFrame到字典转换

Question

我有一个数据帧，我希望将其转换为字典，使col1成为键，col2成为值。此外，第一列中可能出现多次值，在这种情况下，我希望以与col1的值相对应的列表的形式存储计数作为键。 所以基本上我该如何进行这种转换 -

 面积计数

10 7
 10 3
 10 1
 20 5
 30 2
 30 1
 40 3

{
'10'：[7,3,1]，
''20'：'5'，
'30'：[2,1]，
'40'：3
}

提前致谢！

Answer 1

如果我理解你的comment，你可以这样做：

In [35]: df
Out[35]:
   area  count  probability
0    10      7     0.116754
1    10      3     0.703845
2    10      1     0.368201
3    20      5     0.550190
4    30      2     0.037654
5    30      1     0.379534
6    40      3     0.320877

In [36]: df.sort_values('probability', ascending=False).groupby('area')['count'].apply(list).to_dict()
Out[36]: {10: [3, 1, 7], 20: [5], 30: [1, 2], 40: [3]}