Notes
- Get the categorical vars you want to get dummy with.
- use pd.get_dummies to convert one categorical variables to several dummys
1 | catagorical_vars = ['peak','business_line','gender','age_level'] |
2 | continuous_vars = set.difference(set(all_vars),set(catagorical_vars)) |
3 | |
4 | cate_list = [] |
5 | for i in catagorical_vars: |
6 | print(catagorical_vars) |
7 | fe_dummy = pd.get_dummies(X[i]) |
8 | cate_list.append(fe_dummy) |
9 | |
10 | dummy_all = pd.concat(cate_list, axis = 1) |
11 | dummy_all.head() |