熊猫：滚动第二大价值

Question

我需要得到df的滚动第二大值。

以获得最大的价值

max = df.sort_index(ascending=True).rolling(10).max()

当我尝试这个时，python抛出了一个错误

max = df.sort_index(ascending=True).rolling(10).nlargest(2)

AttributeError: 'Rolling' object has no attribute 'nlargest'

这是虫子吗？我还能用什么性能好的？

Answer 1

使用np.sort降序并选择第二个值：

np.random.seed(2019)

df = pd.DataFrame({
    'B': np.random.randint(20, size=15)
})
print (df)
     B
0    8
1   18
2    5
3   15
4   12
5   10
6   16
7   16
8    7
9    5
10  19
11  12
12  16
13  18
14   5

a = df.rolling(10).apply(lambda x: -np.sort(-x)[1]) 
#alternative
#a = df.rolling(10).apply(lambda x: np.sort(x)[-2]) 
print (a)
       B
0    NaN
1    NaN
2    NaN
3    NaN
4    NaN
5    NaN
6    NaN
7    NaN
8    NaN
9   16.0
10  18.0
11  16.0
12  16.0
13  18.0
14  18.0

Answer 2

我会这样做：

df.rolling(10).apply(lambda x: pd.Series(x).nlargest(2).iloc[-1])