将分组依据添加回数据框

Question

我有一个数据框在熊猫如下，这是存储球队的名字赢得和体育场的名字，它赢得了如下

d={"Winner":["Team1","Team2","Team3","Team1","Team2","Team1"],
         "Stadium":["S1","S2","S3","S1","S2","S1"]}

score=pd.DataFrame.from_dict(d)

现在，我在上面的数据框中使用group by进行了一些计算，以找出每个队的计数以及在体育场赢得的比赛，如下所示：

 score.groupby(by=["Stadium","Winner"]).size().reset_index()

现在，我想将这些计数值添加回我的数据帧

我面临多个分组列的问题

任何帮助

Answer 1

这就是merge的意义：

score.merge(
    score.groupby(by=["Stadium","Winner"]).size().reset_index(),
    on=["Stadium", "Winner"]
)