将数据帧转换为字典

时间:2016-12-27 15:08:08

标签: python pandas dataframe

我想将数据帧转换为键和值列表的字典。 索引probe_fk为键,box_fk为值。

             product_fk   vp_fk         box_fk
probe_fk                                   
'83348'            101    326            8548
'83348'            101    326            8549
'83348'            101    326            8550
'83348'            101    326            8551
'105513'           107    357            8866
'105513'           107    357            8864
'111996'           107    365            9059

我希望最终结果如下:

{'83348':[8548, 8549, 8550,8551],
'105513':[8866, 8864],
'111996':[9059]}

1 个答案:

答案 0 :(得分:3)

试试这个:

In [95]: df.groupby(level=0)['box_fk'].apply(list).to_dict()
Out[95]: {'105513': [8866, 8864], '111996': [9059], '83348': [8548, 8549, 8550, 8551]}