Pandas：在滚动window和return的另一列的总和中找到最大值的行并继续四行

Question

我有一个包含两列的数据框。我想找到第一列的滚动 5 周期最大值，并计算滚动最大值行和前 4 行的值和第二列的总和。

下面是所需输出的示例，其中包含如上所述的 Col1 和 Col2、滚动最大值的第三列和所需计算结果的最后一列

这是设置前三列的代码：

data ={'Col1': [4,2,3,4,5,6,5,4,3,2,1,4,3,2,1],
  'Col2' :[10,20,10,15,10,20,10,15,10,20,10,15,10,20,10]}
df = pd.DataFrame(data) 
df['Col1_Rolling5_Max'] = df['Col1'].rolling(5).max()
df

这是所需输出的示例：

Answer 1

您只能计算下一个 Rolling5_Max 不同的行的滚动总和，然后向前填充以填充其余部分。

import pandas as pd
import numpy as np

data ={'Col1': [4,2,3,4,5,6,5,4,3,2,1,4,3,2,1],
  'Col2' :[10,20,10,15,10,20,10,15,10,20,10,15,10,20,10]}
df = pd.DataFrame(data) 
df['Col1_Rolling5_Max'] = df['Col1'].rolling(5).max()


df['flag'] = df['Col1_Rolling5_Max'].ne(df['Col1_Rolling5_Max'].shift())
df['Sum_5_Col1_Before_Max']= np.where(df['flag']==True,df['Col2'].rolling(5).sum(),np.nan)
df.ffill(inplace=True)
df[['Col1','Col2','Col1_Rolling5_Max','Sum_5_Col1_Before_Max']]

输出

    Col1  Col2  Col1_Rolling5_Max  Sum_5_Col1_Before_Max
0      4    10                NaN                    NaN
1      2    20                NaN                    NaN
2      3    10                NaN                    NaN
3      4    15                NaN                    NaN
4      5    10                5.0                   65.0
5      6    20                6.0                   75.0
6      5    10                6.0                   75.0
7      4    15                6.0                   75.0
8      3    10                6.0                   75.0
9      2    20                6.0                   75.0
10     1    10                5.0                   65.0
11     4    15                4.0                   70.0
12     3    10                4.0                   70.0
13     2    20                4.0                   70.0
14     1    10                4.0                   70.0

Pandas：在滚动window和return的另一列的总和中找到最大值的行并继续四行

Pandas: Find max in rolling window and return sum of another column for the row of the max and proceeding four rows

python

dataframe

pandas

rolling-computation