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

Question

rolling_apply在pandas中是否有相当于将函数应用于一系列的累积值而不是滚动值？我意识到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