Python pandas在分组后选择一个行值

Question

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

例子：

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

我想查找县数最多的州，选择州和县只显示，不显示人口。

答案应该是，但我不知道如何在 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. 如果要计算不同的值，请按 STATE 分组，然后使用 nunique 或在 COUNTRY 列上使用 size。
2. 获取计数为最大值的行的索引。