【代码模版】pandas生成哑变量(dummy variable)

注意:生成哑变量时最好采用k类生成k-1个的方法,避免“哑变量陷阱”

# 对于数据框中只含有一个需要哑变量转换的特征:
DataFrame2 = pd.get_dummies(DataFrame1, drop_first=True)
# drop_first = True 保证生成k-1个哑变量
# 对于数据框中多个需要哑变量转化的特征:
DataFrame2 = pd.get_dummies(DataFrame1, columns = ['column_name'], drop_first=True)
# columns : list-like, default None
#     Column names in the DataFrame to be encoded.
#     If `columns` is None then all the columns with
#     `object` or `category` dtype will be converted.
发布了22 篇原创文章 · 获赞 0 · 访问量 932

猜你喜欢

转载自blog.csdn.net/weixin_44680262/article/details/104608778