您就快到了…就像您说的那样,您可以直接添加所有要编码的列
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)



