python - pandas groupby 聚合元素列表加法

Question

我有一个 pandas 数据框，如下所示:

X                      Y
71455  [334.0,  319.0,  298.0,  323.0]
71455  [3.0,  8.0,  13.0,  10.0]
57674  [54.0,  114.0,  124.0,  103.0]

我想执行聚合 groupby 以元素方式添加存储在 Y 列中的列表。我试过的代码:

df.groupby('X').agg({'Y' : sum})   

结果如下:

                                                   Y
X                                                       
71455  [334.0,  319.0,  298.0,  323.0, 75.0,  55.0,  ...

所以它连接了列表，而不是按元素对它们求和。然而，预期的结果是:

X                      Y
71455  [337.0,  327.0,  311.0,  333.0]
57674  [54.0,  114.0,  124.0,  103.0]

我尝试了不同的方法，但无法使它按预期工作。

Answer 1

可以在分组的数据帧上使用apply来获得逐元素的加法:

df.groupby('X')['Y'].apply(lambda x: [sum(y) for y in zip(*x)])

这会产生一个 pandas 系列对象:

X
57674     [54.0, 114.0, 124.0, 103.0]
71455    [337.0, 327.0, 311.0, 333.0]