Acho que vale a pena expandir para o groupby com duas colunas, e talvez também o método .agg, que acrescenta bastante ao uso!