Python 最流行的嵌套类别

我试图用python解决一个问题。我有一个数据帧

df = pd.DataFrame{'company':[1,1,1,1,2,2,3,3,3],'category':['a','a','a','c','a','b','b','b','a']}

我想知道，按公司划分，最受欢迎的类别是什么，其结果是：

对于公司2，它可以是a类或b类，而不是两者，可以随机选择

谢谢

这是

groupby（）

和

模式

：

df.groupby('company', as_index=False)['category'].apply(lambda x: x.mode()[0])

输出：

   company category
0        1        a
1        2        a
2        3        b