python - 当列值在另一行列值的范围内时，Pandas 选择行

Question

我正在尝试从数据帧(100k-500k 行)创建子集格式如下

d = {'time':[1,2,3,5,7,9,9.5,10], 'val':['match','match','match','not','not','match','match','match']}
df = pd.DataFrame(d)

   time    val
0   1.0  match
1   2.0  match
2   3.0  match
3   5.0    not
4   7.0    not
5   9.0  match
6   9.5  match
7  10.0  match

我想选择一个包含时间范围内的所有行的子集范围有限。例如，如果范围为 <=1，则选择第一行和最后三行

我想要的输出:

   time    val
0   1.0  match
1   2.0  match
2   3.0  match
5   9.0  match
6   9.5  match
7  10.0  match

Answer 1

在一行中它看起来像这样:

df.loc[(df['time'].diff()<=1)|(df['time'].diff(-1)>=-1)]