熊貓：組兩列基於其他列的值

我很新的Python /熊貓，我有一個數據幀，看起來是這樣的：由ID熊貓：組兩列基於其他列的值

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   name    color 
id_1  [alex,sara]  [blue,black] 
id_2  [james,aly]  [yellow,red] 
id_4  [dave,lin]  [pink,grey]

有沒有簡單的方法來做到這一點？

來源

2017-08-01 Shushu

的可能的複製https://stackoverflow.com/questions/34776651/concatenate-rows-of-pandas-dataframe-with-same-id – piroot

使用groupby和agg通過自定義函數與tolist：

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]

來源

2017-08-01 11:48:16 jezrael

完美，謝謝！ +1 – Shushu

熊貓：組兩列基於其他列的值

回答

相關問題