如何编辑字典的值?

Question

我有一个包含三个整数列的数据框。我想创建一个字典，它使用前两列作为键，第三列作为值。

为此，我将前两列转换为索引并应用dataframe.to_dict('index')。

问题:我的字典是这样的:{(111,222):{'colname3': 0.1}。

问题:如何从字典的值中编辑列名以将字典转换为:{(111,222): 0.1}?

我的数据框架是这样的:

index colname1 colname2 colname3
0     111      222      0.1
1     111      333      0.2
2     111      444      0.3
3     222      111      0.4
4     222      333      0.5

Answer 1

设置索引为colname1 &2、选择colname3并转换为dict:

df.set_index(['colname1', 'colname2'])['colname3'].to_dict()

输出:

{(111, 222): 0.1,
(111, 333): 0.2,
(111, 444): 0.3,
(222, 111): 0.4,
(222, 333): 0.5}