标签编码
现在来处理分类变量的编码。您将继续使用来自 Kaggle House Prices 比赛的一个子样本。
您的目标是使用标签编码对分类特征 "RoofStyle" 和 "CentralAir" 进行编码。train 和 test 两个 DataFrame 已在您的工作区中可用。
本练习是课程的一部分
用 Python 赢下 Kaggle 竞赛
练习说明
- 将
train和test两个 DataFrame 连接为一个单一的 DataFramehouses。 - 不带任何参数地创建一个
LabelEncoder对象，并将其赋值给le。 - 使用同一个
le对象，为 "RoofStyle" 和 "CentralAir" 创建新的标签编码特征。
交互式实操练习
通过完成这段示例代码来试试这个练习。
# Concatenate train and test together
houses = ____.____([train, test])
# Label encoder
from sklearn.preprocessing import LabelEncoder
le = ____()
# Create new features
houses['RoofStyle_enc'] = le.fit_transform(houses[____])
houses['CentralAir_enc'] = ____.____(____[____])
# Look at new features
print(houses[['RoofStyle', 'RoofStyle_enc', 'CentralAir', 'CentralAir_enc']].head())