python - 将数据框转换为字典

Question

我想将数据帧转换为键字典和值列表。 索引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]}

Answer 1

试试这个:

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