Get numerical and categorical columns

PHOTO EMBED

Mon Sep 04 2023 03:54:41 GMT+0000 (Coordinated Universal Time)

Saved by @aadpaguio

cols = train.drop(columns=['AUTO_F', 'CAR_OWNER' ]).columns
num_cols = train._get_numeric_data().columns
cat = list(set(cols) - set(num_cols))
content_copyCOPY