熊猫滚动应用：应用（）得到了一个意想不到的关键字参数

Question

我想有一个滚动应用的数据帧，但是我有问题与我的自定义函数，我想有一个额外的输入：

df_test = pd.DataFrame(columns=['amount'])
df_test['amount'] = [1, 2, 3, 4, 5]
mean = df_test['amount'].mean()

def rule(x,mean):
   x = x-mean
   return sum(x)

df_test['amount'].rolling(3).apply(rule,mean=mean)

这是回报

TypeError: apply() got an unexpected keyword argument 'mean'

Answer 1

它不同于正常的**kwargs，这里是滚动。apply基本上，它是一个参数，在dict中接受关键字args。

文档<代码>滚动。应用：

>

kwargs：默认为无。

要传递到func的关键字参数。

df_test['amount'].rolling(3).apply(rule, kwargs={'mean':mean})

0    NaN
1    NaN
2   -3.0
3    0.0
4    3.0
Name: amount, dtype: float64

还有一种方法是使用functools。此处为部分。

from functools import partial
rule_mean = partial(rule, mean=mean)
df_test['amount'].rolling(3).apply(rule_mean)

0    NaN
1    NaN
2   -3.0
3    0.0
4    3.0
Name: amount, dtype: float64