python - 如何为 pandas groupby 设计 agg 函数

Question

我的数据框是这样的:

user,rating, f1,f2,f3,f4

20, 3, 0.1, 0, 3, 5

20, 4, 0.2, 3, 5, 2

18, 4, 0.6, 8, 7, 2

18, 1, 0.7, 9, 2, 7

例如，我想为用户计算一个配置文件

对于用户 20，它应该是 3*[0.1,0,3,5]+4*[0.2,3,5,2]
这是 f1 到 f4

的加权和

我应该如何编写一个agg函数来完成这个任务？

df.groupby('user').agg(....)

Answer 1

你可以试试这个:

df.groupby('user').apply(lambda x : sum(x['rating'] * (x['f1']+x['f2']+x['f3']+x['f4'])))