python - 熊猫滚动最小最大

Question

如何在数据帧上大小为 n 的滚动窗口中找出 max 和 min 中的第一个并存储量化值（如果 max 先出现，则说 1，如果 min 先出现，则说 0）？

Answer 1

你可以尝试这样的事情：

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