“Jak wybrać 5 najlepszych w każdej grupie pandy” Kod odpowiedzi

Jak wybrać 5 najlepszych w każdej grupie pandy

df.groupby('id').apply(lambda x : x.sort_values(by = 'value', ascending = False).head(2).reset_index(drop = True))

#Alter native

df.groupby(['id']).apply(lambda x: x.nlargest(topk,['value'])).reset_index(drop=True)