python - 评级标准化

Question

我有下一个 pandas DataFrame df:

user item rating
1    1    1
1    2    1
1    3    3
2    1    2
2    2    2
2    3    1
...

我想要规范化评分，定义0到1之间的所有评分值。方法很简单，只需将某个用户的每个值除以该用户的最大值即可。

我创建了下一个代码:

ratingNormalised = []

for user in df['user'].unique:
    dfUser = df[df['user'] == user]
    userNormalised = (dfUser['rating']/max(dfUser['rating'])).tolist()
    ratingNormalised.extend(userNormalised)

df['ratingNorm'] = Series(ratingNormalised, index=df.index)

是否有可能找到一些更好的解决方案，可能更Pythonic？

Answer 1

对用户进行分组并应用 lambda:

In [73]:

df['norm rating'] = df.groupby('user')['rating'].apply(lambda x: x/x.max())
df

Out[73]:
   user  item  rating  norm rating
0     1     1       1     0.333333
1     1     2       1     0.333333
2     1     3       3     1.000000
3     2     1       2     1.000000
4     2     2       2     1.000000
5     2     3       1     0.500000