python - 浏览和选择 2 个数据框

Question

假设我有 2 个像这样的 DataFrame:

df1
     Id     B
3    alpha  1
18   alpha  3
125  alpha  5
230  alpha  9

其中 3, 18, 125, 230 是 df1 的索引

和:

df2
     Id     B
1    Beta  21
2    Beta  33
5    Beta  120
7    Beta  36
10   Beta  32
14   Beta  71
15   Beta  210
21   Beta  53
123  Beta  22 
127  Beta  1227
128  Beta  11
227  Beta  7
235  Beta  18 

我的问题:如何浏览我的 df2 以根据索引提取 df1 每一行前后的行？

我想要的输出:

df2
     Id     B
2    Beta  33
5    Beta  120
15   Beta  210
21   Beta  53
123  Beta  22 
127  Beta  1227
227  Beta  7
235  Beta  18 

Answer 1

使用searchsorted对于位置，添加一个先前的值，加入 numpy.r_并选择 iloc :

a = df2.index.searchsorted(df1.index) 
df = df2.iloc[sorted(np.r_[a, a-1])]
print (df)
       Id     B
2    Beta    33
5    Beta   120
15   Beta   210
21   Beta    53
123  Beta    22
127  Beta  1227
227  Beta     7
235  Beta    18