熊貓：合並DataFrame中的列

Question

以下面的DataFrame為例：

import pandas as pd
df = pd.DataFrame({'col1':[1, 2, 3, 2, 1] , 'col2':['A', 'A', 'B', 'B','C']})

   col1 col2
0     1    A
1     2    A
2     3    B
3     2    B
4     1    C

我怎樣才能得到

   col1  col2  
0   1     A,C  
1   2     A,B
2   3     B  

Answer 1

您可以對'col1'進行groupby ，然后apply連接值的lambda：

In [88]:
df = pd.DataFrame({'col1':[1, 2, 3, 2, 1] , 'col2':['A', 'A', 'B', 'B','C']})
df.groupby('col1')['col2'].apply(lambda x: ','.join(x)).reset_index()

Out[88]:
   col1 col2
0     1  A,C
1     2  A,B
2     3    B