Zastosuj funkcję na zbiorczych wartościach serii pand

Question

Czy istnieje odpowiednik rolling_apply w pandach, który stosuje funkcję do skumulowanych wartości szeregu zamiast wartości toczenia? Rozumiem, że istnieją: cumsum, i cummin, ale chciałbym zastosować funkcję niestandardową.

Answer 1

Można użyć pd.expanding_apply. Poniżej znajduje się prosty przykład, który tak naprawdę zawiera skumulowaną sumę, ale możesz napisać, jaka jest dla ciebie funkcja.

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