python - 过滤 df 中的行并在字符串值之间返回 - pandas

Question

以下返回一个 bool 值，其中整数是前一行的增加。

但是，我希望结合基于字符串的过滤行。具体来说，我想返回 B 之间的行和 C .所以每当C关注 B

import pandas as pd

df = pd.DataFrame({                   
    'X' : ['A','B','C','D'],
    'Y' : [1.0,1.0,2.0,1.0],              
   })

df = df.assign(Z = df.Y.diff().gt(0))

出去:

     X      Y
0  1.0  False
1  1.0  False
2  2.0   True
3  1.0  False

但是我想在 C 时使用字符串值来返回行关注 B .

预期:

   X    Y
1  B  3.0
2  C  2.0

Answer 1

是你想要的吗:

# current is B and next is C
s = df['X'].eq('B') & df['X'].shift(-1).eq('C')

# current row and the next row
df[s | s.shift()]