Question

以下是我的数据框。我正在尝试计算ATR的滚动5个周期百分比等级。 RollingPercentRank是我想要的输出。

       symbol         Day      time       ATR  RollingPercentRank
316356    SPY  11/29/2018  10:35:00  0.377880                 NaN
316357    SPY  11/29/2018  10:40:00  0.391092                 NaN
316358    SPY  11/29/2018  10:45:00  0.392983                 NaN
316359    SPY  11/29/2018  10:50:00  0.399685                 NaN
316360    SPY  11/29/2018  10:55:00  0.392716                 0.2
316361    SPY  11/29/2018  11:00:00  0.381445                 0.2
316362   AAPL  11/29/2018  11:05:00  0.387300                 NaN
316363   AAPL  11/29/2018  11:10:00  0.390570                 NaN
316364   AAPL  11/29/2018  11:15:00  0.381313                 NaN
316365   AAPL  11/29/2018  11:20:00  0.398182                 NaN
316366   AAPL  11/29/2018  11:25:00  0.377364                 0.6
316367   AAPL  11/29/2018  11:30:00  0.373627                 0.2

从第5行开始，我想将百分比等级函数应用于组中ATR的所有5个先前值（第1行至第5行）。从第6行开始，我想再次对ATR的所有5个先前值（第2行到第6行）应用rank函数。我尝试了以下给出“ numpy.ndarray”对象没有属性“ rank”的错误。

df['RollingPercentRank'] = df.groupby(['symbol'])['ATR'].rolling(window=5,min_periods=5,center=False).apply(lambda x: x.rank(pct=True)).reset_index(drop=True)

Answer 1

IIUC，因为我没有得到您所显示的预期输出，但是要使用rank，您需要一个pd.Series，然后您只需要此百分比的5个元素序列的最后一个值，因此将是：

print (df.groupby(['symbol'])['ATR']
         .rolling(window=5,min_periods=5,center=False)
         .apply(lambda x: pd.Series(x).rank(pct=True).iloc[-1]))

symbol  i     
AAPL    316362    NaN
        316363    NaN
        316364    NaN
        316365    NaN
        316366    0.2
        316367    0.2
SPY     316356    NaN
        316357    NaN
        316358    NaN
        316359    NaN
        316360    0.6
        316361    0.2

由于x组成一个numpy数组，因此可以使用twice argsort获得相同的结果，并在最后创建一个reset_index列：

win_val = 5
df['RollingPercentRank'] = (df.groupby(['symbol'])['ATR']
                              .rolling(window=win_val,min_periods=5,center=False)
                              .apply(lambda x: x.argsort().argsort()[-1]+1)
                              .reset_index(level=0,drop=True)/win_val)

print (df)
       symbol         Day      time       ATR  RollingPercentRank

316356    SPY  11/29/2018  10:35:00  0.377880                 NaN
316357    SPY  11/29/2018  10:40:00  0.391092                 NaN
316358    SPY  11/29/2018  10:45:00  0.392983                 NaN
316359    SPY  11/29/2018  10:50:00  0.399685                 NaN
316360    SPY  11/29/2018  10:55:00  0.392716                 0.6
316361    SPY  11/29/2018  11:00:00  0.381445                 0.2
316362   AAPL  11/29/2018  11:05:00  0.387300                 NaN
316363   AAPL  11/29/2018  11:10:00  0.390570                 NaN
316364   AAPL  11/29/2018  11:15:00  0.381313                 NaN
316365   AAPL  11/29/2018  11:20:00  0.398182                 NaN
316366   AAPL  11/29/2018  11:25:00  0.377364                 0.2
316367   AAPL  11/29/2018  11:30:00  0.373627                 0.2

熊猫的滚动百分比排名

1 个答案: