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

由于city 1 person 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'})
)