パンダ：グループ別の列

Question

の値に基づいて、2つの列Iのpython /パンダにかなり新たなんだと私はこのような何かに見えるデータフレームがあります。

id   name  color 
id_1  alex  blue 
id_2  james  yellow 
id_1  sara  black 
id_4  dave  pink 
id_4  lin   grey 
id_2  aly   red 

を私はidでグループ化すると取得他の2つの列の値をリストとして表示します。

id   name    color 
id_1  [alex,sara]  [blue,black] 
id_2  [james,aly]  [yellow,red] 
id_4  [dave,lin]  [pink,grey] 

これを行う簡単な方法はありますか？ tolistとカスタム関数によって

Answer 1

使用groupbyとagg：

df = df.groupby('id').agg(lambda x: x.tolist()) 
print (df) 
       name   color 
id        
id_1 [alex, sara] [blue, black] 
id_2 [james, aly] [yellow, red] 
id_4 [dave, lin] [pink, grey]