Pandas Dataframe группирует и фильтрует по макс (ключ)

Question

td = [[10,'A'],[10, 'B'],[2, 'C']]
import pandas
df = pandas.DataFrame(td, columns['col1','col2'])
df.groupby('col1')

Все, как мне получить группу по макс (клавиша). В этом случае 10 -> ['A', 'B']?

Спасибо! * * 1004

Answer 1

Обобщенное решение возможно через groupby для перечисления, затем используя sort_index и iloc accessor:

res = df.groupby('col1')['col2']\
        .apply(list)\
        .sort_index(ascending=False)

print(res.iloc[0])

['A', 'B']