将列表中的行与多列分组

Question

我在pandas中有这个东西：

    a   b   c
0   A   1   6
1   A   2   5
2   B   3   4
3   B   4   3
4   B   5   2
5   C   6   1

我想将其转换为此：

    a   b          c
0   A   [1, 2]     [6, 5]
1   B   [3, 4, 5]  [4, 3, 3]
2   C   [6]        [1]

最有效的方法是什么？

Answer 1

好的，就是这样

df = df.groupby('a').agg({'b': list, 'c':list}).reset_index()

如果有什么更好的话，您可以告诉我。