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]
1
例子:
import pandas as pd
df = pd.DataFrame([  
            ['green' , 'A'],   
            ['red'   , 'B'],   
            ['blue'  , 'A']])  

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

get_dummies 前: 
 
get_dummies 後: 


上述執行完以後再打印df 出來的還是get_dummies 前的圖,因爲你沒有寫

df = pd.get_dummies(df)

可以對指定列進行get_dummies
pd.get_dummies(df.color)


將指定列進行get_dummies 後合併到元數據中
df = df.join(pd.get_dummies(df.color))

--------------------- 
 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章