python-3.x - 来自OneHotEncoder的功能名称

我正在使用OneHotEncoder编码一些分类变量(例如-Sex和AgeGroup)。编码器产生的特征名称类似-'x0_female'，'x0_male'，'x1_0.0'，'x1_15.0'等。

>>> train_X = pd.DataFrame({'Sex':['male', 'female']*3, 'AgeGroup':[0,15,30,45,60,75]})

>>> from sklearn.preprocessing import OneHotEncoder
>>> encoder = OneHotEncoder()
>>> train_X_encoded = encoder.fit_transform(train_X[['Sex', 'AgeGroup']])

>>> encoder.get_feature_names()
>>> array(['x0_female', 'x0_male', 'x1_0.0', 'x1_15.0', 'x1_30.0', 'x1_45.0',
       'x1_60.0', 'x1_75.0'], dtype=object)

有没有一种方法可以告诉OneHotEncoder创建特征名称，以便在开头添加列名称，类似于-Sex_female，AgeGroup_15.0等，类似于 Pandas get_dummies()所做的。

最佳答案

您可以将具有原始列名的列表传递给get_feature_names:

encoder.get_feature_names(['Sex', 'AgeGroup'])

将返回:

['Sex_female', 'Sex_male', 'AgeGroup_0', 'AgeGroup_15',
 'AgeGroup_30', 'AgeGroup_45', 'AgeGroup_60', 'AgeGroup_75']

关于python-3.x - 来自OneHotEncoder的功能名称，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/54570947/