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]}

关于python - 枢纽分析表到字典，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/29685386/