python - 在 pandas 中创建一个分数列，其值取决于另一列的百分位数

Question

我有以下数据框:

User_ID Game_ID votes
1         11    1040
1         11    nan
1         22    1101
1         11    540
1         33    nan
2         33    nan
2         33    290
2         33    nan

根据 votes 列中值的百分位数，需要按照以下规则创建一个新列:

If the “votes” value is >= 75th percentile assign a score of 2

If >=25th percentile assign a score of 1

If <25th percentile assign a score of 0.

Answer 1

您可以通过调用 describe 和使用列表理解来获取百分位数:

percentiles = df.votes.describe()
df['scores'] = [2 if x >= percentiles['75%'] else (0 if x < percentiles['25%'] else 1) for x in df.votes]