Найдите два верхних элемента (тип Stage Gate, тип Int) для каждой группы («GUID»)

Question

У меня есть фрейм данных, который я уже отсортировал по группе / «GUID».И я хочу выбрать только 2 верхних «Stage Gate» для каждого GUID.

Я пытался:

df.groupby('GUID')['Stage Gate'].nlargest(2)

, но я получаю ту же ошибку: не могу использовать метод 'nlargest' с dtypeобъект

obj = dataProcess(df)
obj.df.sort_values(by=['GUID','Stage Gate'], ascending=False).apply(lambda d: d.nlargest(2))

Answer 1

Вы можете сделать с sort_values + tail

df.sort_values('Stage Gate').groupby('GUID').tail(2)