Pandas按多列分组，并将结果作为新列放入地图

Question

我有一个格式为

的数据框

|Col1|Col2|Count|Day  |
|----|----|-----|-----|
| A  |  C |  1  | Mon |
| A  |  C |  3  | Tue |
| A  |  C |  5  | Wed |
| A  |  D |  7  | Mon |
| B  |  C |  2  | Mon |
| B  |  D |  4  | Tue |

我希望按前两列进行分组，并创建一个每日计数字典作为分组原始行的新列，并丢弃原始的Count和Day列。我希望我的结果看起来像

|Col1|Col2|dateCounts                   |
|----|----|-----------------------------|
| A  |  C |{"Mon":1, "Tue": 3, "Wed": 5}|
| A  |  D |{"Mon":7}                    | 
| B  |  C |{"Mon":2}                    |
| B  |  D |{"Tue":4}                    |

我可以做一些事情，例如将列添加到每一行，删除Count和Day列，然后删除框架上的重复项，但我很难获得额外的dateCounts列以便正常工作。这可以使用df.groupby(['Col1', 'Col2']).apply()方法吗？

Answer 1

使用groupby + apply：

df.groupby(['Col1', 'Col2'])\
   .apply(lambda x: dict(zip(x.Day, x.Count))).reset_index(name='dateCounts')

  Col1 Col2                      dateCounts
0    A    C  {'Wed': 5, 'Tue': 3, 'Mon': 1}
1    A    D                      {'Mon': 7}
2    B    C                      {'Mon': 2}
3    B    D                      {'Tue': 4}