在熊猫数据框中按字符值保留组

问题描述

import pandas as pd

df = pd.DataFrame({'A': [1,1,2,2],'B': ['sam','anu','rita','first','mid','last']})
print(df)

我有上面的数据框，我想转换如下

非常感谢任何帮助。谢谢！

解决方法

试试

out = df.groupby('A',as_index=False).agg(tuple)
   A                   B
0  1    (sam,anu,rita)
1  2  (first,mid,last)