Pandas - 將多行數據合並到單行的語法

Question

我正在嘗試將多行中的數據合並到具有多列的單行中。 當我想要多列時，我只能管理一列。

下面給出的是數據的樣子：

column1 column2 column3
1       a       Y
1       b       Z
2       d       Z
2       c       Y

預期 output：

column1 column2 column3
1       a, b       Y, Z
2       d, c       Z, Y

使用df.groupby('column1')['column2'].apply(', '.join).reset_index()我得到了 output：

column1 column2 
1       a, b      
2       d, c    

但不確定添加 column3 的語法。

df.groupby('column1')['column2', 'column3'].apply(', '.join).reset_index()不起作用。

df.groupby('column1')[['column2', 'column3']].apply(', '.join).reset_index() 。

Answer 1

您可以嘗試agg進行多個連接：

df.groupby('column1').agg(lambda x: ', '.join(x)).reset_index()