Data-Science-REU/OneHotEncodingData.py at main · amosr2002/Data-Science-REU · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
# -*- coding: utf-8 -*-
"""
Created on Wed Jun  2 19:34:53 2021

@author: balle
"""


import pandas as pd

df = pd.read_csv('Cleaned_Cancer_Behavior_Dataset.csv')
df = df.drop(['Unnamed: 0', 'AgeGroup'], axis=1)

dummies = pd.get_dummies(df.Gender).iloc[:,1:]
dummies2 = pd.get_dummies(df.Cancer).iloc[:,1:]


df = pd.concat([df, dummies, dummies2], axis=1)
df = df.drop(['Cancer', 'Gender'], axis=1)


#df = df.rename(columns={ df.columns[81]: "drinking_1"})
#df = df.rename(columns={ df.columns[82]: "drinking_2"})
#df = df.rename(columns={ df.columns[81]: "employed_1"})

dummiesNew = pd.get_dummies(df.Employed).iloc[:,1:]
df = df.drop(['Employed'], axis=1)
df = pd.concat([df, dummiesNew], axis=1)
df = df.rename(columns={ df.columns[80]: "Employed_Patient"})
df = df.rename(columns={ df.columns[81]: "Unemployed_Patient"})
df = df.rename(columns={ df.columns[77]: "Gender"})

df = df.drop(['Unemployed_Patient'], axis=1)

df.to_csv('Data_Cleaned_Cancer_Data.csv')