pandas.get_dummies 的用法

get_dummies 是利用pandas实现one hot encode的方式。详细参数请查看官方文档
官方文档在这里

pandas.get_dummies(data, prefix=None, prefix_sep='_', dummy_na=False, columns=None, sparse=False, drop_first=False)[source]

例子:

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

df.columns = ['color',  'class'] 
pd.get_dummies(df) 

get_dummies 前:
pandas.get_dummies 的用法_第1张图片
get_dummies 后:
pandas.get_dummies 的用法_第2张图片

上述执行完以后再打印df 出来的还是get_dummies 前的图,因为你没有写

df = pd.get_dummies(df)

可以对指定列进行get_dummies

pd.get_dummies(df.color)

pandas.get_dummies 的用法_第3张图片

将指定列进行get_dummies 后合并到元数据中

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

pandas.get_dummies 的用法_第4张图片

你可能感兴趣的:(pandas,python)