# This gets hard to read very quickly
df = (
    df
    .assign(...)
    .loc[...]
    .groupby(...)
    .agg(...)
    .reset_index()
    .rename(...)
    .sort_values(...)
    .query(...)
)