python groupby 然后一些列先保留其他列最后

Question

我有一个看起来像的数据框:

cityid    personid    yearstart monthstart yearend monthend  
1          1           2000       01        2001    02
1          1           2001       02        2001    10
1          2           2001       10        2002    10
2          3           2000       01        2002    12
2          4           2005       08        2006    12

自 person 1在 city 1连续有两个术语，我想合并这两行并得到:

cityid    personid    yearstart monthstart yearend monthend  
1          1           2000       01        2001    10
1          2           2001       10        2002    10
2          3           2000       01        2002    12
2          4           2005       08        2006    12

所以每一行都有一个唯一的键 {cityid, personid}。
我试过df = df.groupby['cityid','personid'].['yearstart','momthstart'].first()['yearend, monthend'].last()但收到错误消息。
我能问一下如何解决这个问题吗？谢谢!

Answer 1

您可以使用 agg :

(df.groupby(['cityid','persionid'])
   .agg({'yearstart':'first',
         'monthstart':'first',
         'yearend':'last',
         'monthend':'last'})
)