如何选择特定列中与给定值保持一定距离的所有数据帧行？

Question

下面是一个示例数据帧，我将使用它来更好地说明我的问题：

import pandas as pd

df = pd.DataFrame(pd.np.random.rand(30, 3), columns=tuple('ABC'))
df['event'] = pd.np.nan
df.loc[10, 'event'] = 'ping'
df.loc[20, 'event'] = 'ping'
df.loc[19, 'event'] = 'pong'

我需要创建以ping为中心的n行的窗口。在

换句话说，让i是event列中包含ping的行的索引。对于每个i，我想选择df.ix[i-n:i+n]。在

因此，对于n=3，我期望得到以下结果：

^{pr2}$

一些注意事项：

1. 我在寻找一个非迭代的解决方案。在
2. 请注意，有一个pong值，我们不想将窗口置于中心位置。但是，它是在第二个ping中心的结果中捕获的。在

如何做到这一点？在

Answer 1

也许：

>>> ts, n = df['event'] == 'ping', 3
>>> idx = ts.shift(n).fillna(False)  # +n rows
>>> for j in range(-n, n):  # -n to n-1 rows
...     idx |= ts.shift(j).fillna(False)
... 
>>> df[idx]