在 Pandas 滚动时使用带有参数的自定义函数

Question

为了说明问题

import pandas as pd
import numpy as np    

def testfunc(x, mult=1):
    res = sum(x)*mult
    return res

series = pd.Series(np.random.rand(100), name='test')

我想在大小为 10 的滚动窗口上运行“testfunc”，例如，使用 mult 参数的“系列”。

series.rolling(10).apply(testfunc, raw=False, **{'mult':2})

如果我尝试上述方法，我会收到错误

TypeError: apply() 得到了一个意外的关键字参数 'mult'

从帮助页面看来，这应该是可能的，但我找不到工作示例

Answer 1

根据this doc，Series.rolling().apply() 不接受**kwargs。取而代之的是 kwargs 选项，它需要一个字典：

series.rolling(10).apply(testfunc, raw=False, kwargs={'mult':2})

输出：

0           NaN
1           NaN
2           NaN
3           NaN
4           NaN
        ...    
95    11.782115
96    10.999794
97     9.678652
98     9.669550
99    10.093348
Name: test, Length: 100, dtype: float64