熊貓數據框分組依據並通過max（key）進行過濾

Question

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

全部，如何通過max（key）獲取組。 在這種情況下10-> ['A'，'B']？

謝謝！

Answer 1

通用解決方案可以通過groupby列出，然后使用sort_index和iloc訪問器：

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

print(res.iloc[0])

['A', 'B']