python - 找到等于的运行平均值

Question

假设我有一些与足球相关的数据

Date   Home     Away  HomeGoal AwayGoal TotalGoal
2019   Arsenal  MU     5        1        6
2019   MCity    Liv    2        2        4
2019   MU       Liv    3        4        7
2019   MCity    MU     0        0        0

我想创建一列数据来显示该球队最近两场比赛的平均进球数。例如，在最后一行中，我想包含一列，显示 MU 在最近 2 场比赛中的平均进球数，即 = (1+3)/2 = 2。

python中有没有函数可以实现这个功能？

Answer 1

对于您的要求，您不关心球队是主队还是客队，只关心每场比赛进了多少球。试试这个:

# Rename the columns to make the unstacking operation a bit easier
# Always a good idea to specify an explicit `copy` when you intend
# to change the dataframe structure
>>> tmp = df[['Home', 'Away', 'HomeGoal', 'AwayGoal']].copy()

# Arrange the columns into a MultiIndex to make stacking easier
>>> tmp.columns = pd.MultiIndex.from_product([['Team', 'Goal'], ['Home', 'Away']])

# This is what `tmp` look like:

           Team      Goal     
      Home Away Home Away
0  Arsenal   MU    5    1
1    MCity  Liv    2    2
2       MU  Liv    3    4
3    MCity   MU    0    0

# And now the magic
>>> tmp.stack() \
        .groupby('Team').rolling(2).mean() \
        .groupby('Team').tail(1) \
        .droplevel([1,2])

# Result
         Goal
Team         
Arsenal   NaN
Liv       3.0
MCity     1.0
MU        1.5

其工作原理如下:

• stack 取消透视 Home 和 Away，这样对于每场比赛，我们都有 2 行用于 Teams 和 目标
• groupby('Team').rolling(2).mean() 获取每支球队最近 2 场比赛的滚动平均进球数
• groupby('Team').tail(1) 获取每个团队的最后一个滚动平均值
• 此时，过渡数据帧的索引有 3 个级别:球队名称、比赛编号和上一场比赛的主/客场指示符。我们只关心第一个，所以我们会放弃另外两个。