display all categorical column in pandas dataframe
# get all categorical columns in the dataframe
catCols = [col for col in df.columns if df[col].dtype=="O"]
display all categorical column in pandas dataframe
# get all categorical columns in the dataframe
catCols = [col for col in df.columns if df[col].dtype=="O"]
To convert categorical data to numerical
cat_cols = ['Item_Identifier', 'Item_Fat_Content', 'Item_Type', 'Outlet_Identifier',
'Outlet_Size', 'Outlet_Location_Type', 'Outlet_Type', 'Item_Type_Combined']
enc = LabelEncoder()
for col in cat_cols:
train[col] = train[col].astype('str')
test[col] = test[col].astype('str')
train[col] = enc.fit_transform(train[col])
test[col] = enc.transform(test[col])
Copyright © 2021 Codeinu
Forgot your account's password or having trouble logging into your Account? Don't worry, we'll help you to get back your account. Enter your email address and we'll send you a recovery link to reset your password. If you are experiencing problems resetting your password contact us