python - 在 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