熊猫滚动最大值

时间:2018-06-13 12:13:44

标签: python pandas rolling-computation

如何在数据帧的大小为n的滚动窗口中查找最大值和最小值之间的值并存储量化值(如果最大值为1则为1,如果最小值为0则为0)?

1 个答案:

答案 0 :(得分: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