python之数据透视表到字典

我有这个数据透视表:

[in]:unit_d 
 
[out]: 
                         units 
store_nbr   item_nbr     
   1            9        27396 
                28        4893 
                40        254 
                47        2409 
                51        925 
                89        157 
                93        1103 
                99        492 
 
    2           5         55104 
                11        655 
                44        117125 
                85        106 
                93        653 

我想要一个以“store_nbr”为键，以“item_nbr”为值的字典。
所以，{'1': [9, 28, 40,...,99], '2': [5, 11 ,44, 85, 93], ...}

请您参考如下方法:

我会使用 groupby在这里，在重置索引以使其成为列之后:

>>> d = unit_d.reset_index() 
>>> {k: v.tolist() for k, v in d.groupby("store_nbr")["item_nbr"]} 
{1: [9, 28, 40, 47, 51, 89, 93, 99], 2: [5, 11, 44, 85, 93]}