获取 Pandas 中最后一次出现特定值后的所有行

Question

我的数据框看起来像

ID  colA  

1      B     

1      D     

2      B    

2      D     

2      C     

我已经在每组中最后一次发生事件 B 后返回所有行。输出将是：

ID  colA   

1      D   

2      D     

2      C  

我试过

a = df['colA'].str.contains('B').groupby(df['ID'])

b = df[(a.transform('sum') - a.cumsum()).eq(0)]

到目前为止它运行良好。我只是想知道是否有其他方法可以实现这一目标？

Answer 1

国际大学联盟def yourlast(x):        return x.loc[x.colA.where(df.colA.eq('B')).last_valid_index()+1:]df.groupby('ID').apply(yourlast)Out[163]:       ID colAID           1  1   1    D2  3   2    D   4   2    C

Answer 2

你可以这样做：ix = (df.colA.eq('B')        .cumsum()        .groupby(df.ID)        .apply(lambda x: x.loc[x.idxmax()+1:]).index.get_level_values(1))df.loc[ix,:]    ID colA1   1    D3   2    D4   2    C

Answer 3

反转您的行（这很重要）。然后调用groupbyand cumsum，并取（反向）cumsum 值等于零的所有行。df[df.colA.eq('B')[::-1].astype(int).groupby(df.ID).cumsum().eq(0)]   ID colA1   1    D3   2    D4   2    C