我试图在Pandas中一起使用groupby,nlargest和sum功能,但是无法使它工作.
State County Population
Alabama a 100
Alabama b 50
Alabama c 40
Alabama d 5
Alabama e 1
...
Wyoming a.51 180
Wyoming b.51 150
Wyoming c.51 56
Wyoming d.51 5
Run Code Online (Sandbox Code Playgroud)
我想用groupby州选择,然后按人口排名前2个县.然后只使用前2个县人口数来获得该州的总和.
最后,我将有一个列表,其中包含州和人口(前2个县).
我可以得到groupby并且nlargest工作,但获得总和nlargest(2)是一个挑战.
我现在的路线很简单: df.groupby('State')['Population'].nlargest(2)