# Categorical boolean mask categorical_feature_mask = X.dtypes==object # filter categorical columns using mask and turn it into a list categorical_cols = X.columns[categorical_feature_mask].tolist()