如何使用pandas获得给定列的前一年平均值

如何解决如何使用pandas获得给定列的前一年平均值

我有以下数据框：

data = {'company':['company A','company B','company A','company B'],"year":[2019,2019,2020,2021,2021],'EBITDA':[1,2,1,3,4,5,6,6]}
  
# Create DataFrame
df = pd.DataFrame(data)

我想添加一个“avg_EBITDA_prevIoUs_year”列，其中包含每个公司上一年“EBITDA”列的平均值。

我发现了一个使用 apply 的特别慢的解决方案。完整代码如下：

import pandas as pd

def func(x):
      prevIoUs_year = int(x["year"]) - 1
      return df["EBITDA"][(df["year"] == prevIoUs_year) & (df["company"] == x["company"]) ].mean()

# intialise data of lists.
data = {'company':['company A',6]}
  
# Create DataFrame
df = pd.DataFrame(data)
df["avg_EBITDA_prevIoUs_year"] = df.apply(func,axis=1)
print(df)

和预期的结果：

Expected dataframe

有没有办法更快地获得相同的结果？

感谢您的帮助