熊貓合併行

Question

我有一個熊貓DataFrame，其中一些行對具有相同的ID但名稱不同。我想要的是將行對減少到一行，並顯示它們的名字。

INPUT：

ID NAME  AGE 
149 Bob  32 
150 Tom  53 
150 Roberts 53 
151 Pamela 28 
152 Andrew 23 

輸出：

ID NAME  AGE 
149 Bob   32 
150 Tom Roberts 53 
151 Pamela  58 
152 Andrew  23 

否則，我還可以做[ '湯姆'， '羅伯茨']，或仍然捕獲數據的任何其它方法。

Answer 1

用groupby很容易完成。

df = df.groupby('ID', as_index=False)\ 
     .agg({'NAME' : ' '.join, 'AGE' : 'first'}) 

print(df)  
ID   NAME AGE 
149   Bob 32 
150 Tom Roberts 53 
151  Pamela 28 
152  Andrew 23