如何在groupby条件满足后返回索引？

Question

我有一个很大的df.pandas，我想创建一个函数来对一个列进行分组并计算另一列中的uniqe值，然后返回最大计数值的索引，如下所示：

def groupby_fun():
    x = df.groupby(by=df['Column1'])['Column2'].nunique()
    return [x.max()].index[0]
groupby_fun()

我收到此错误：

'builtin_function_or_method' object is not subscriptable

我做错了什么，拜托？

Answer 1

添加.idxmax()作为Nickil Maveli建议工作：

df.groupby('Column1')['Column2'].nunique().idxmax()