反向排序应用Python Pandas指数加权平均值

问题描述 投票:2回答:2

对于pandas.DataFrame示例:

In: cols = ['cols1', 'cols2']
In: df = pd.DataFrame({'col1': [1, 2, 3, 4], 'col2': [3, 4, 5, 6]})

Out:       col1  col2
      0     1     3
      1     2     4
      2     3     5
      3     4     6

我使用的是指数加权平均值:

In: for i in range(len(df.columns)):
       df[cols[i]] = df[cols[i]].ewm(com=None, span=None, halflife=None, 
                                     alpha=.8, min_periods=0, adjust=True,
                                     ignore_na=False, axis=0).mean()

它很棒!但是,权重从上到下应用指数递减的值:

  Out:      col1      col2
      0  1.000000  3.000000
      1  1.833333  3.833333
      2  2.774194  4.774194
      3  3.756410  5.756410

我想知道是否有一种方法以相反的顺序(自下而上)应用权重。我想要的输出是:

Out:          col1          col2
      0     0.9391025     2.8173075
      1     1.8494627     3.6982925
      2     2.7499995     4.5833325
      3     4.000000      6.000000
python pandas exponential weighted weighted-average
2个回答
1
投票

我找到了问题的解决方案。您可以在应用ewm()之前和之后重新索引:

In: df.reindex(index=df.index[:-1])
    for i in range(len(df.columns)):
        df[cols[i]] = df[cols[i]].ewm(com=None, span=None, halflife=None, 
                                      alpha=.8, min_periods=0, adjust=True,
                                      ignore_na=False, axis=0).mean()
Out:        col1      col2
      3  4.000000  6.000000
      2  3.166667  5.166667
      1  2.225806  4.225806
      0  1.243590  3.243590

然后你可以再次申请:

In: df.reindex(index=df.index[:-1])
Out:        col1      col2
       0  1.243590  3.243590
       1  2.225806  4.225806
       2  3.166667  5.166667
       3  4.000000  6.000000

我仍然不确定这是否是最有效的方法。所以,如果有人有任何其他想法,我很想知道。


0
投票

kel's solution的变种。

注意

df[col].reindex(index=df.index[::-1])

达到类似的结果:

df[col].sort_index(ascending=False)

一个区别是reindex改变了指数,而sort_index则选择inplace。因此,与sort_index不同,reindex的一个优点是你不需要在最后切换回来:

df[col] = df[col].sort_index(ascending=False).ewm(...).mean()

另请注意,使用sort_index的解决方案当然假定索引值以正确的方式排序。如果他们不是,我不确定是否采取滚动平均值将具有物理意义。

© www.soinside.com 2019 - 2024. All rights reserved.