在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值的百分位數，需要根據以下規則創建新列：

如果“投票”值> = 75thcentntile，則得分為2

如果> = 25th百分位，則分數為1

如果<25thcentntile指定得分為0。

Answer 1

使用pd.qcut ：

df['score'] = pd.qcut(df['votes'].astype(float), [0, 0.25, 0.75, 1.0]).cat.codes
print(df)

輸出（ nan對應於-1 ）：

0    1
1   -1
2    2
3    1
4   -1
5   -1
6    0
7   -1
dtype: int8

Answer 2

您可以通過調用describe並使用list comprehension來獲取百分位數：

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