python - 根据上一个和下一个值过滤 Pandas 数据框

Question

我的数据框有 3 列 [C1-C3]

C1  C2  C3
a   h   1
b   i   0
c   j   0
d   k   1
e   l   1
f   m   1
g   n   0

我有兴趣根据有关上一个和下一个值的标准过滤其行 具体来说，我想保留所有行，其中 C3 具有模式 0-1-1 (previous_value = 0 和 next_value = 1)。因此，在上面使用的示例中，数据框应转换为:

C1  C2  C3
d   k   1

Answer 1

使用 shift

m = df.C3.shift(1).eq(1) & df.C3.eq(1) & df.C3.shift(-1).eq(1)
0    False
1    False
2    False
3    False
4     True
5    False
6    False
Name: C3, dtype: bool
df = df[m]