python - 检查 pandas 数据帧的最后一行是否满足条件的最佳方法是什么？

Question

我在尝试创建一个新列时遇到了困难，该新列是基于“信号”列的检查列。如果最后五行(包括最后一行)为 1，则返回 1，如果最后五行(包括最后一行)为 0，则返回 0，其他均为 check 的最后一个值，如下所示:

我有以下数据框:

       signal
index        
0           1
1           1
2           1
3           1
4           1
5           1
6           0
7           0
8           0
9           0
10          0
11          0
12          0
13          1
14          0
15          1
16          1
17          1
18          1
19          1

我想要这样的东西:

       signal  check
index                
0           1       1
1           1       1
2           1       1
3           1       1
4           1       1
5           1       1
6           0       1
7           0       1
8           0       1
9           0       1
10          0       0
11          0       0
12          0       0
13          1       0
14          0       0
15          1       0
16          1       0
17          1       0
18          1       0
19          1       1

我将不胜感激任何形式的帮助!

谢谢!

Answer 1

尝试rolling.sum对df和ffill、bfill进行切片

df['check'] = df[df.rolling(5).sum().isin([0, 5])].ffill().bfill()

Out[540]:
       signal  check
index
0           1    1.0
1           1    1.0
2           1    1.0
3           1    1.0
4           1    1.0
5           1    1.0
6           0    1.0
7           0    1.0
8           0    1.0
9           0    1.0
10          0    0.0
11          0    0.0
12          0    0.0
13          1    0.0
14          0    0.0
15          1    0.0
16          1    0.0
17          1    0.0
18          1    0.0
19          1    1.0

如果您希望检查为整数，只需链加astype(int)

df['check'] = df[df.rolling(5).sum().isin([0, 5])].ffill().bfill().astype(int)