python数据帧到字典，键值问题

Question

我正在创建一个数据框，然后转换为字典，如下所示

data = {'ID': [1,2,3,4,5],
       'A':['1','2','1','3','2'],
       'B':[4,6,8,2,4]}
frame = pd.DataFrame(data)

dict_obj = dict(frame[['A','B']].groupby('A').median().sort_values(by='B'))

我的问题是我想要列A作为键，列B作为值但不知何故我得到一个奇怪的字典

dict_obj
{'B': A
 3    2
 2    5
 1    6
 Name: B, dtype: int64}

我希望字典对象为

{1:6,2:5,3:2}

有人可以帮忙吗？

Answer 1

使用pd.Series.to_dict方法

frame.groupby('A').B.median().to_dict()

{'1': 6, '2': 5, '3': 2}