OneHotEncoding 分类数据

Question

我正在尝试使用 OneHotEncoding 转换我的 csv 文件的第二列，其中包含公司名称。

from sklearn.preprocessing import OneHotEncoder
ct = ColumnTransformer(transformers=[('encoder', OneHotEncoder(), [1])]), remainder='passthrough'
X = np.array(ct.fit_transform(X))

我收到了这个错误：

    ct = ColumnTransformer(transformers=[('encoder', OneHotEncoder(), [1])]), remainder='passthrough'
                  ^
SyntaxError: can't assign to function call

我哪里错了？我知道我写的ct行可能有错误，但我不知道我到底做错了什么。

Answer 1

我觉得你设置的括号不对，大概应该是这样的：

ct = ColumnTransformer(transformers=[('encoder', OneHotEncoder(), [1])], remainder='passthrough')

Answer 2

你可以先从sklearn.compose调用columntransformer 像这样写代码来自 sklearn.compose 导入 ColumnTransformer 来自 sklearn.preprocessing 导入 OneHotEncoder ct = ColumnTransformer(transformers=[('encoder', OneHotEncoder(), [1])]), remainder='passthrough' X = np.array(ct.fit_transform(X))

OneHotEncoding 分类数据

OneHotEncoding Categorical data

python

neural-network

scikit-learn

one-hot-encoding