python - 如何连接具有公共(public)列值的行？

Question

例如:

Index   Variable     Name
0         'var1'    'apple'
1         'var2'    'orange'
2         'var1'    'pineapple'

我想得到这个:

Index    Variable     Name
0         'var1'      ['apple','pineapple']
1         'var2'      ['orange']

我尝试使用 df.groupby(['Variable']) 但我不知道如何将它转换回数据框。

Answer 1

您可以在分组对象上应用list函数进行聚合，这会将每个组中的Name Series对象变成一个列表，并将其视为单个元素结果:

df.groupby("Variable").Name.apply(list).reset_index()