如何使用列转换器对多个列进行编码?

时间:2019-10-31 08:21:39

标签: python machine-learning scikit-learn

我想用columtransformer编码多个列。你知道怎么添加吗?

from sklearn.preprocessing import LabelEncoder
from sklearn.compose import make_column_transformer
transformer = make_column_transformer(( OneHotEncoder(categories='auto'), [1] ),remainder="passthrough")
X = transformer.fit_transform(X)

LabelEncoder_Y = LabelEncoder()
y = LabelEncoder_Y.fit_transform(y)

image

1 个答案:

答案 0 :(得分:0)

对于make_column_transformer,您需要编码转换的列的索引/列名称的list。例如,如果您需要列索引0和1:

transformer = make_column_transformer( (OneHotEncoder(categories='auto'), 
                                       [0, 1]), remainder="passthrough" )