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))
---------------------