python - 合并行并将某些列放在行尾而不是制作列表

Question

我想知道list在执行类似groupby().

我想要做的是dataframe这样的：

index NAME  ID  COLOR
0   Tim     1   red
1   Sally   2   red
2   Rob     3   green
3   Sally   2   blue
4   Sally   2   purple

然后合并行来做这个：

index NAME  ID  COLOR1  COLOR2  COLOR3 
0   Tim     1   red
1   Sally   2   red     blue    purple
2   Rob     3   green

我知道如何agg()使用，groupby()但我试图弄清楚在字段中追加而不是list在字段中创建，这样dataframe它将是list字段中的COLOR。

标签： pythonpandasdataframe