Панды: консолидация столбцов в 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

Вы можете groupby на «col1», а затем apply лямбда, который присоединяется значения:

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