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

Question

这个问题已经有答案了:

How to group dataframe rows into list in pandas groupby (16 个回答)

已关闭 4 年前。

这是一个例子:

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  Y
1  a          [cat, dog]
2  b    [hi, hello, hey]
Name: Z, dtype: object

编辑:可以仅应用list而不是Series.tolist，如@timegb在评论中建议的那样。