将累积平均函数应用于分组对象答案

【问题标题】：Applying cumulative mean function to a grouped object将累积平均函数应用于分组对象
【发布时间】：2014-01-14 02:47:50
【问题描述】：

我有一个 DataFrame df，其中每条记录代表一场足球比赛。团队将不止一次出现。我需要为每个团队的分数计算某种滚动平均值（嗯，不完全是滚动平均值）。

     date           home           away       score_h  score_a
166  2013-09-01     Fulham         Chelsea       0      0
167  2013-09-03     Arsenal        Everton       0      2
164  2013-09-05     Arsenal        Swansea       5      1
165  2013-09-06     Fulham         Norwich       0      1
163  2013-09-18     Arsenal        Swansea       0      0

我需要计算的是每支球队（主客场）的平均得分。

为简洁起见，我们只做首页栏目：

grouped = df.groupby('home')
grouped = grouped.sort_index(by='date') # rows inside groups must be in asc order

这会导致：

    date    home    away    score_h     score_a
home                        
Arsenal     167     2013-09-03  Arsenal     Everton     0   2
            164     2013-09-05  Arsenal     Swansea     5   1
            163     2013-09-18  Arsenal     Swansea     0   0
Fulham      166     2013-09-01  Fulham      Chelsea     0   0
            165     2013-09-06  Fulham      Norwich     0   1

问题从这里开始

现在，我需要计算团队的“滚动平均值”。让我们为名为@987654325@ 的组手动操作。最后，我们应该得到 2 个额外的列，我们称它们为：rmean_h 和 rmean_a。组中的第一个记录 (167) 的分数为 0 和 2。其中的rmean 分别是0 和2。对于组中的第二条记录 (164)，rmeans 将是 (0+5)/2 = 2.5 和 (2+1) / 2 = 1.5，对于第三条记录，(0+5+0)/3 = 1.66 和 (2+1+0)/3 = 1。

我们的 DataFrame 现在应该如下所示：

                    date       home         away    score_h score_a rmean_h rmean_a
home                        
Arsenal     167     2013-09-03  Arsenal     Everton     0  2    0       2
            164     2013-09-05  Arsenal     Swansea     5  1    2.5     1.5
            163     2013-09-18  Arsenal     Swansea     0  0    1.66    1
Fulham      166     2013-09-01  Fulham      Chelsea     0  0
            165     2013-09-06  Fulham      Norwich     0  1

我想对我的数据进行这些计算，请问您有什么建议吗？

【问题讨论】：

标签： python group-by pandas

【解决方案1】：

您可以将expanding_mean（参见docs）应用于每个组：

grouped = df.sort(columns='date').groupby('home')
grouped['score_h'].apply(pd.expanding_mean)

【讨论】：

如果您想要问题中的结果，您需要将home 设置为索引（groupby 不会为您执行此操作）。例如。类似df = df.set_index('home', drop=False, append=True).reorder_levels([1,0]).sort_index()
不需要 lambda；你可以直接.apply(pd.expanding_mean)。
@DSM 啊，确实如此。谢谢！我调整了我的答案。