對熊貓系列的累積值應用函數

Question

在rolling_apply中是否有等效的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_)