python - 根据多行中的值在 DataFrame 中选择行

Question

我有一个 DataFrame，我想选择在一列中具有特定值的行，并且上面的行在另一列中具有特定值。在没有 for 循环的情况下如何做到这一点？

例如:

df = pd.DataFrame({'one': [1,2,3,4,1,2,3,4], 'two': [1,2,3,4,5,6,7,8]})

我想在其中查找该行上的 df.one 等于 1 且该行上的 df.two 的行上面等于4，因此在示例中第4行的值为[1,5]。

Answer 1

您可以尝试shift与 boolean indexing :

print df
   one  two
0    1    1
1    2    2
2    3    3
3    4    4
4    1    5
5    2    6
6    3    7
7    4    8

print (df.one == 1) & (df.two.shift() == 4)
0    False
1    False
2    False
3    False
4     True
5    False
6    False
7    False
dtype: bool

print df[(df.one == 1) & (df.two.shift() == 4)]
   one  two
4    1    5