如何在groupby之后聚合熊猫中的向量列?

问题是 x 是一个pd.Series的numpy.arrays,假设你想要质心,你可以使用 np.vstack 并找到平均值的第一轴:

安装程序

import numpy as np
import pandas as pd

arrays = [np.array([-0.030460168, -0.0014596573, 0.0997446, -0.18]),
          np.array([-0.03197706, 0.015620711, 0.05890667, -0.0402]),
          np.array([-0.060918115, 0.07939958, 0.0333591, 0.035655]),
          np.array([-0.012854534, 0.07733478, 0.12939823, 0.00992]),
          np.array([-0.04184026, 0.03382166, 0.1427004, -0.204424])]

users = ['bob', 'bob', 'bob', 'mary', 'mary']

df = pd.DataFrame(data={'user': users, 'features': arrays})

代码

result = df.groupby('user').agg(count=('user', lambda x: len(x)),
                       mean=('features', lambda x: np.vstack(x).mean(axis=0).tolist()))

print(result)

输出

      count                                               mean
user                                                          
bob       3  [-0.04111844766666667, 0.031186877899999996, 0...
mary      2  [-0.027347397, 0.055578220000000005, 0.1360493...