Python Pandas Dataframe select row by max value in group

Question

I have a dataframe which was created via a df.pivot:

type                             start  end
F_Type         to_date                     
A              2         


        

Answer 1

The other ways to do that are as follow:

1. If you want only one max row per group.

(
    df
    .groupby(level=0)
    .apply(lambda group: group.nlargest(1, columns='to_date'))
    .reset_index(level=-1, drop=True)
)

2. If you want to get all rows that are equal to max per group.

(
    df
    .groupby(level=0)
    .apply(lambda group: group.loc[group['to_date'] == group['to_date'].max()])
    .reset_index(level=-1, drop=True)
)