基于组将多行合并到一个单元格中

Question

我有这样的csv文件格式：

user1 , group1
user2 , group1
user3 , group2
user4 , group2

我想使用python来获得以下结果：

group1, user1 user2
group2, user3 user4

我该怎么做？

Answer 1

使用groupby和lambda函数连接文本字符串。

import pandas as pd
df = pd.DataFrame({'user': {0: 'user1', 1: 'user2', 2: 'user3', 3: 'user4'}, 
'group': {0: 'group1', 1: 'group1', 2: 'group2', 3: 'group2'}})
df.groupby('group')['user'].apply(lambda x: ' '.join(x))

group
group1    user1 user2
group2    user3 user4
Name: user, dtype: object