lable encoding a column code example
Example 1: python convert categorical data to one-hot encoding
df_onehot = pd.get_dummies(df, columns=['col_name'], prefix=['one_hot'])
df = pd.DataFrame(['sunny', 'rainy', 'cloudy'], columns=['weather'])
print(df)
weather
0 sunny
1 rainy
2 cloudy
df_onehot = pd.get_dummies(df, columns=['weather'], prefix=['one_hot'])
print(df_onehot)
one_hot_cloudy one_hot_rainy one_hot_sunny
0 0 0 1
1 0 1 0
2 1 0 0
Example 2: label encoding
from sklearn.preprocessing import LabelEncoder
le = LabelEncoder()
companydata.ShelveLoc = le.fit_transform(companydata.ShelveLoc)