Python - 不确定如何将列中的行值汇总到列表中

Question

下面是一个例子：

df1：

X Y Z

1 a cat

1 a dog

2 b hi

2 b hello

2 b hey

最终的 df 应该是这样的。

df2：

X Y Z

1 a [cat, dog]

2 b [hi, hello, hey]

我真的被卡住了，我什至无法接近这个。任何帮助将非常感激。

Answer 1

您可以使用groupby和apply：df.groupby(['X', 'Y'])['Z'].apply(list)返回X  Y1  a          [cat, dog]2  b    [hi, hello, hey]Name: Z, dtype: object编辑：可以只申请list而不是Series.tolist@timegb 在评论中建议的那样。