将分类数字数据编码到不同的列

Question

我是数据科学和机器学习的新手。我有一列包含三个值0,1,2，并且我想将这三个值编码为3个带有前缀predict_的不同列。

我尝试过get_dummies和标签编码器，但没有锻炼

import pandas as pd
Y = pd.get_dummies(Y,prefix='predict_')

from sklearn.preprocessing import LabelEncoder
le = LabelEncoder()
Y = le.fit_transform(Y)

最后，列的外观应为dict_0，predict_1，predict_2

Answer 1

确保仅传递列

pd.get_dummies(Y['column_name'],prefix='predict_')