python - 从 if else 语句创建新列

Question

我有一个名为 data 的数据框，其薪资范围可变

SALARY:0 1 2 3 ...  1000

我想对工资变量进行存储并创建一个新的可变工资范围。其中，如果工资= 0，则缺少带，1-200 为低，200-800 为中，800 以上为高。

data['SALARY_BAND'] = np.where(data['SALARY']<=0, 'Missing', 
                             np.where(data['SALARY'] <=200, 'low'),
                              np.where(data['SALARY'] <800, 'mid','high' ))

当我运行时，我得到错误值缺少 y

Answer 1

我会使用cut函数:

df = pd.DataFrame(list(range(0, 1001, 10)), columns=['SALARY'])
df['SALARY_BAND'] = pd.cut(df.SALARY,
                           [-1, 0, 200, 800, df.SALARY.max()],
                           labels=['Missing', 'low', 'mid', 'high'])