Pandas 滚动最小最大值

Pandas rolling min max

如何在数据帧上大小为 n 的滚动 window 中找出最大和最小值中排在第一位的那个并存储量化值(如果最大值在前则为 1,如果最小值在前则为 0)?

您可以尝试这样的操作:

np.random.seed(124)
df = pd.DataFrame({'data':np.random.randint(0,10,25)})
df['First'] = df['data'].rolling(5, min_periods=2).apply(lambda x: x.idxmin()<x.idxmax(), raw=False)
df['First'] = df['First'].map({1:'Min',0:'Max'})
df

输出:

    data First
0      1   NaN
1      7   Min
2      2   Min
3      9   Min
4      0   Max
5      4   Max
6      4   Max
7      5   Max
8      5   Min
9      6   Min
10     9   Min
11     6   Min
12     0   Max
13     8   Max
14     9   Max
15     9   Min
16     0   Min
17     2   Max
18     2   Max
19     1   Max
20     1   Min
21     9   Min
22     1   Min
23     9   Min
24     4   Min