我想在数据框中找到一栏的移动平均值,我需要定义一个具有4个参数的函数:数据框,国家/地区,列,窗口以计算移动平均值。这是我的代码:
def movingaverage (values, country, column, window):
df = values[values['Countries'] == 'country']
s = pd.Series(df['column'])
sma = s.rolling(window).median
return sma
SMA_5 = movingaverage (values=df, country='Albania', column='data', window=5)
但是当我调用该函数时,它说:KeyError: 'column'
这是我的数据框的一部分:
Year ISO_Code Countries Economic Freedom Summary Index Rank Quartile Government
2017 Albania ... ... ...
2017 Algeria
...
我的函数必须具有4个参数。谢谢!