Python学习笔记——pandas中get_dummies()的用法

原文链接在此
get_dummies()函数默认转了以后的变量用columns名_数值名表示

import pandas as pd
df = pd.DataFrame([
    ['green', 'A'],
    ['red', 'B'],
    ['blue', 'A']
])
df.columns = ['color', 'class']
pd.get_dummies(df)

可以得到以下输出

其中参数predix：
prefix : 给输出的列添加前缀，如prefix=“A”,输出的列会显示类似
prefix_sep : 设置前缀跟分类的分隔符sepration，默认是下划线"_"

如：

pd.get_dummies(df,prefix_sep='@')

如果是将指定列进行get_dummies 后合并到元数据中，使用如下指令:

df = df.join(pd.get_dummies(df.color))
df