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

Question

import pandas as pd

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

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

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

Answer 1

试试

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