python - Pandas Dataframe中的焊盘选择范围？

Question

如果我用类似的东西切片数据框

>>> df = pd.DataFrame(data=[[x] for x in [1,2,3,5,1,3,2,1,1,4,5,6]], columns=['A'])

>>> df.loc[df['A'] == 1]
# or
>>> df[df['A'] == 1]

   A
0  1
4  1
7  1
8  1

我怎样才能用 1 的缓冲区填充我的选择并获取每个索引0, 1, 3, 4, 5, 6, 7, 8, 9？我想选择列中'A'值为 1 的所有行，以及任何此类行之前或之后的行。

---

编辑我希望找到一种适用于任意焊盘尺寸的解决方案，而不仅仅是适用于 1 的焊盘尺寸。

---

编辑 2这是另一个示例，说明我要做什么

df = pd.DataFrame(data=[[x] for x in [1,2,3,5,3,2,1,1,4,5,6,0,0,3,1,2,4,5]], columns=['A'])

我们正在寻找pad == 2. 在这种情况下，我会尝试获取 rows 0, 1, 2, 4, 5, 6, 7, 8, 9, 12, 13, 14, 15, 16。

标签： pythonpandasdataframe

Answer 1

您可以按位使用 shift 或|

c = df['A'] == 1
df[c|c.shift()|c.shift(-1)]

---

   A
0  1
1  2
3  5
4  1
5  3
6  2
7  1
8  1
9  4