python - 对 Pandas 系列的累积值应用函数

Question

pandas 中是否有等效的 rolling_apply 将函数应用于一系列的累积值而不是滚动值？我意识到 cumsum、cumprod、cummax 和 cummin 存在，但我想应用自定义函数.

Answer 1

您可以使用 pd.expanding_apply .下面是一个简单的示例，它只真正计算累加和，但您可以为它编写任何您想要的函数。

import pandas as pd

df = pd.DataFrame({'data':[10*i for i in range(0,10)]})

def sum_(x):
    return sum(x)


df['example'] = pd.expanding_apply(df['data'], sum_)

print(df)

#   data  example
#0     0        0
#1    10       10
#2    20       30
#3    30       60
#4    40      100
#5    50      150
#6    60      210
#7    70      280
#8    80      360
#9    90      450