在Pandas dataframe中使用OneHotEncoder后,可以通过以下步骤将encoder.categories_添加为列名:
import pandas as pd
from sklearn.preprocessing import OneHotEncoder
data = {'category': ['A', 'B', 'C', 'A', 'B']}
df = pd.DataFrame(data)
encoder = OneHotEncoder()
encoded_data = encoder.fit_transform(df[['category']])
categories = encoder.categories_[0]
df_encoded = pd.DataFrame(encoded_data.toarray(), columns=categories)
df_final = pd.concat([df, df_encoded], axis=1)
最终,df_final将是一个包含原始数据列和使用OneHotEncoder编码后的新列的Pandas dataframe。其中,新列的列名将使用encoder.categories_中的类别列表。
关于OneHotEncoder的概念、分类、优势和应用场景,以及腾讯云相关产品和产品介绍链接地址,可以参考腾讯云的文档和官方网站。
领取专属 10元无门槛券
手把手带您无忧上云