Groupby id 包含列表的列

Question

假设您有以下数据框：

>>> df = pd.DataFrame({'id':[0,0,1], 'lst':[["a", "b", "c"],["e"],["c"]]})
>>> df
  id         lst
0   0  [a, b, c]
1   0        [e]
2   1        [c]

您希望按 id 对数据进行分组，如下所示：

   id        lst
0   0  [[a, b, c], [e]]
2   1  [[c]]

有没有办法对列表的列进行分组，所以可以获得我上面给出的结果？

我尝试了以下链接中发布的解决方案：Group operations on Pandas column containing lists，但它对我不起作用。

Answer 1

这是一种方法

# groupby id, then then apply list to the resulting grouped lst items

df.groupby('id')['lst'].apply(list).reset_index()

    id  lst
0   0   [[a, b, c], [e]]
1   1   [[c]]