对熊猫系列的累积值应用函数

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

Answer 2

[关注@Ffisegydd的回答]

熊猫更新== 1.0.5

df['example'] = df['data'].expanding().apply(sum_)