如何加入具有公共列值的行？

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()