python - 如何在列中获取真值和侧翼假值

Question

我想知道如何最简单地选择一列中的所有 True 行以及 Trues 集合中紧邻的 Falses 行。

TLDR; Extract all rows that are True, along with the False rows 1 above, and 1 below every True rows.

这就是我的意思:

df:

0     False
1     False
2      True
3      True
4     False
5     False
6     False
7     False
8      True
9      True
10     True
11    False
12    False
13    False
14    False
15     True
16     True
dtype: bool

然后，运行代码后，结果将是:

结果:

1   False
2   True
3   True
4   False
7   False
8   True
9   True
10  True
11  False
14  False
15  True
16  True

Answer 1

我认为这可以满足需要:

import pandas as pd

df = pd.DataFrame({'value': [
    False,
    False,
    True,
    True,
    False,
    False,
    False,
    False,
    True,
    True,
    True,
    False,
    False,
    False,
    False,
    True,
    True,
]})

result = df[df.value | df.value.shift(-1) | df.value.shift(1)]
print(result)

输出:

    value
1   False
2    True
3    True
4   False
7   False
8    True
9    True
10   True
11  False
14  False
15   True
16   True