如何根据分组值将字典插入数据帧的行中？

Question

我有以下数据：

                                 count
Year         Month  Code    
2016         1      ENE001SOLC   121
                    SAL016DECL   92
                    TRN002SIGN   54
                    HAB002SOLC   38
                    TRE001SIGN   37
                    ESP003SOLC   36
                    TRN002SOLC   32
             2      ENE001SOLC   151
                    CAT001SOLC   143
                    VIT001SOLC   90
                    TRE001SIGN   80
                    TRN002SOLC   74
                    BOM001SOLC   72
             3      ENE001SOLC   114
                    ENT002JUST   96
                    TRE001SIGN   94
                    TRN002SIGN   89
                    ENT002APOR   76
                    TRN002SOLC   56

此数据样本按年份和月份分组。它显示给定代码的频率，如您所见

我想做的是，把同一个月出现的所有代码值，放到python字典中，其中键是代码，计数是值，这样我就可以在同一行中，所有键和它们各自的计数。所需的输出应如下所示：

Year         Month  Code_count    
2016         1      {ENE001SOLC: 121, SAL016DECL: 92, TRN002SIGN: 54, HAB002SOLC: 38, HAB002SOLC: 38, TRE001SIGN: 37, ESP003SOLC: 36, TRN002SOLC: 32}

             2      {ENE001SOLC: 151, CAT001SOLC: 143, VIT001SOLC: 90, TRE001SIGN: 80, TRN002SOLC: 74, BOM001SOLC: 72}

你可以看到我的想法。我曾经尝试过使用pd.to_dict()方法，但是通过这种方法，我可以用它们的代码获得所有的代码值，但是没有考虑分组。我怎么能这么做？？任何帮助都将不胜感激

事先非常感谢

Answer 1

尝试以下方法

result = (df.reset_index()
            .groupby(['Year','Month'])
            .apply(lambda x:x.set_index('Code')['count'].to_dict())
         )