Python允许在group by之后选择行值

我可以知道如何在按列分组后选择具有最大计数的值行吗

示例：

STATE COUNTY POPULATION
1     5571   1000
2     3421   2000
3     6781   3000
2     1234   4000
2     3344   6600   
1     5566   9900

我想找到count of county的最大数量的州，选择仅显示的州和县，不显示人口

答案应该是，但我不知道如何用python编写代码。谢谢你的帮助

STATE   COUNTY
2       3

尝试：

---

v:

国家/地区0 2 3

方法：

按州分组，然后使用

nunique

计算不同的值，或者在国家/地区列中使用

size

获取count为最大值的行的索引

提示：

df.groupby（'STATE'）['COUNTRY'].nunique（）

u = df.groupby('STATE')['COUNTRY'].size()
v = u[u.index==u.idxmax()].reset_index()