Python大熊猫分组后选择行值

Question

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

示例：

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

我想找到最多县数的STATE，选择STATE和COUNTY仅显示，而没有POPULATION。

答案应该是，但是我不知道如何在python中编码。感谢您的帮助

STATE   COUNTY
2       3

Answer 1

尝试：

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

---

v：

STATE COUNTRY 0 2 3

方法：

1. 按州分组，然后如果要计数不同的值，请使用nunique，或在“国家/地区”列中使用size。
2. 获取其中count为最大值的行的索引。