将 Table 旋转到词典

Question

我有这个支点table:

[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], ...}

Answer 1

我会在这里使用 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]}