在 Pandas groupby 上應用 ewm 函數

Question

我有這個數據幀

ddd = pd.DataFrame({'day':['m','t','m','t','m','t','m','t'] 
                    ,'val':[15,2,10,4,5,8,2,10]})

我想按day 分組並將val與ewm聚合

如果我做 groupby 並且意味着我得到了這個

ddd.groupby(['day']).mean()

     val
day     
m      8
t      6

現在就像這樣，如果我嘗試

ddd.groupby(['day']).ewm(com=1.0).mean()

我收到錯誤AttributeError: 'DataFrameGroupBy' object has no attribute 'ewm'

我的預期輸出看起來像這樣

     val
day     
m      4.733333
t      8.133333

Answer 1

並非所有dataframe或series函數都為groupby對象實現。 但是您可以使用apply來獲得預期的結果

ddd.groupby(['day']).apply(lambda x: x.ewm(com=1.0).mean().iloc[-1])

出去：

          val
day          
m    4.733333
t    8.133333