text_plain_output_1.png
import pandas as pd # data processing, CSV file I/O (e.g. pd.read_csv)

df = pd.read_csv('train.csv')

df.isnull().sum()

cats = df.dtypes == 'object'
object_cols = list(cats[cats].index)

cat_features = [cat_val for cat_val in df.columns if 'cat' in cat_val]
print(cat_features)
num_cols = [col for col in df.columns if 'cont' in col]
print(num_cols)