时间戳

首页 > 解决方案 > 在一列中给定相同的值，连接剩余的行？

python - 在一列中给定相同的值，连接剩余的行？

问题描述

给定熊猫数据框：

name   hobby   since
paul   A       1995 
john   A       2005 
paul   B       2015
mary   G       2013
chris  E       2005
chris  D       2001
paul   C       1986

我想得到：

name   hobby1   since1    hobby2   since2    hobby3   since3
paul   A        1995      B        2015      C        1986 
john   A        2005      NaN      NaN       NaN      NaN
mary   G        2013      NaN      NaN       NaN      NaN
chris  E        2005      D        2001      NaN      NaN

即我想每个名字有一行。一个人可以拥有的最大爱好数量，在这种情况下说 3，是我事先知道的。最优雅/最简洁的方法是什么？

标签： pythonpandas

解决方案

您可以先melt，然后，groupby.cumcount()添加到variable，然后使用pivot_table()：

m=df.melt('name')
(m.assign(variable=m.variable+(m.groupby(['name','variable']).cumcount()+1).astype(str))
 .pivot_table(index='name',columns='variable',values='value',aggfunc='first')
.rename_axis(None,axis=1))

      hobby1 hobby2 hobby3 since1 since2 since3
name                                           
chris      E      D    NaN   2005   2001    NaN
john       A    NaN    NaN   2005    NaN    NaN
mary       G    NaN    NaN   2013    NaN    NaN
paul       A      B      C   1995   2015   1986

推荐阅读