python - 如何在Python中找到特定长度的数据帧中的最大值？

Question

这是我的数据框:

DATE_TIME
1997-02-25 12:50:00    3238.0
1997-02-25 12:55:00    3237.5
1997-02-25 13:00:00    3237.5
1997-02-25 13:05:00    3237.5
1997-02-25 13:10:00    3239.0
1997-02-25 13:15:00    3242.0

我正在尝试找到一种方法来找到每 2 个间隔的最大值，并且仍然保持 DataFrame 格式。

所以我希望它看起来像这样。

DATE_TIME
1997-02-25 12:50:00    3238.0
1997-02-25 12:55:00    3238.0
1997-02-25 13:00:00    3238.0
1997-02-25 13:05:00    3237.5
1997-02-25 13:10:00    3239.0
1997-02-25 13:15:00    3242.0

我尝试过各种不同的事情。任何帮助将不胜感激。

Answer 1

您可以使用.rolling().max()方法:

In [141]: df
Out[141]:
            DATE_TIME     VAL
0 1997-02-25 12:50:00  3238.0
1 1997-02-25 12:55:00  3237.5
2 1997-02-25 13:00:00  3237.5
3 1997-02-25 13:05:00  3237.5
4 1997-02-25 13:10:00  3239.0
5 1997-02-25 13:15:00  3242.0

In [142]: df['roll_3_max'] = df['VAL'].rolling(3).max().bfill()

In [143]: df
Out[143]:
            DATE_TIME     VAL  roll_3_max
0 1997-02-25 12:50:00  3238.0      3238.0
1 1997-02-25 12:55:00  3237.5      3238.0
2 1997-02-25 13:00:00  3237.5      3238.0
3 1997-02-25 13:05:00  3237.5      3237.5
4 1997-02-25 13:10:00  3239.0      3239.0
5 1997-02-25 13:15:00  3242.0      3242.0

说明:

In [140]: df['VAL'].rolling(3).max()
Out[140]:
0       NaN
1       NaN
2    3238.0
3    3237.5
4    3239.0
5    3242.0
Name: VAL, dtype: float64