如何在python中将多值变量转换为整数[duplicate]

Question

如果我有一个带有两个值的变量（例如Sex可以取男性或女性），我会使用如下代码：，

train_df["Sex"] = train_df["Sex"].apply(lambda sex: 0 if sex == 'male' else 1)

将字符串转换为整数。如果变量包含两个以上的值，如分类为低/中/高的工资，该怎么做？如何分配与上述类似的值？

Answer 1

您可以进行转换判决，例如：

values = {
    "low" : 0,
    "med" : 1,
    "high": 2
}
train_df["Sex"] = train_df["Sex"].apply(lambda level: values.get(level, 0))

Answer 2

通过字典使用map：

d = {
    'male': 0,
    'female': 1,
    'other': 2
}

train_df["Sex"] = train_df["Sex"].map(d)

但对于薪资而言，如果需要按范围设置新值，则最好削减：

train_df = pd.DataFrame({'Salary': [100,200,300,500]})


bins = [0, 200, 400, np.inf]
labels=['low','medium','high']
train_df['label'] = pd.cut(train_df['Salary'], bins=bins, labels=labels)
print (train_df)
   Salary   label
0     100     low
1     200     low
2     300  medium
3     500    high