Home > Blockchain >  How to find rows from a groupby condition in original dataframe?
How to find rows from a groupby condition in original dataframe?

Time:11-22

I have the following condition which eliminates the last row of a group if that last row does not equal 'no' for the column outcome:

m1 = df.groupby(['id'])['outcome'].tail(1) != 'no' 

I then use this condition to drop these rows from the dataframe:

df = df.drop(m1[m1].index) 

However I do not know how to do the opposite and instead of dropping these rows from the original df, extract entire rows that satisfy the m1 condition. Any suggestions?

CodePudding user response:

From the comments:

df.loc[m1[m1].index, :]

will work.

  • Related