Pandas Dataframe groupby并按max(键)过滤

时间:2018-05-02 18:52:39

标签: python pandas dataframe

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

全部,如何通过max(键)获取组。在这种情况下,10 - > [' A'' B']?

谢谢!

1 个答案:

答案 0 :(得分:0)

可以通过groupby列出广告解决方案,然后使用sort_indexiloc访问者:

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

print(res.iloc[0])

['A', 'B']