一键编码sklearn中的多个列和命名列

您就快到了…就像您说的那样，您可以直接添加所有要编码的列

fit_transform

ohe = oneHotEnprer(categories='auto')feature_arr = ohe.fit_transform(df[['phone','city']]).toarray()feature_labels = ohe.categories_

然后，您只需要执行以下 *** 作：

feature_labels = np.array(feature_labels).ravel()

这使您可以根据需要命名列：

features = pd.Dataframe(feature_arr, columns=feature_labels)