python - 使用 OneHotEncoder 从数据框中删除标签

Question

在包含分类变量的列上使用OneHotEncoder会从整个数据框中删除标签。有什么解决办法吗？我不想丢失我的数据框的标签。

labelencoder_1 = LabelEncoder()
df1['MSZoning'] = labelencoder_1.fit_transform(df1['MSZoning'])
onehotencoder = OneHotEncoder(categorical_features = [2])
df1 = onehotencoder.fit_transform(df1).toarray()

标签： pythonmachine-learningscikit-learnone-hot-encodingdummy-variable