Get a List of Non-Numeric Data Type Columns by Index

Python

Returns a list of data frame columns by index that are object data types. Useful for defining categorical columns when training a Catboost model.

 1|  from pandas.api.types import is_numeric_dtype   
 2|  def get_categorical_indicies(X):  
 3|          cats = []
 4|          for col in X.columns:
 5|              if is_numeric_dtype(X[col]):
 6|                  pass
 7|              else:
 8|                  cats.append(col)
 9|          cat_indicies = []
10|          for col in cats:
11|              cat_indicies.append(X.columns.get_loc(col))
12|          return cat_indicies
13|  categorical_indicies = get_categorical_indicies(X)
detro
Catboost | Pandas
Did you find this snippet useful?

Sign up for free to to add this to your code library

Remove Stop Words from Text in DataFrame Column
Python

Pandas | Nltk | Nlp

149
LightGBM Custom Loss Function
Python

LightGBM | Loss function

132
Tuning XGBoost Hyperparameters with Grid Search
Python

XGBoost | Tuning

127
How to Convert DataFrame Values Into Percentages
Python

Lambda | Percentages | Pipe

119
How to Scale Data Using Standard Scaler But Keep Column Names
Python

Scaler | Standard

92
LightGBM Hyperparameter Tuning with GridSearch
Python

Python | LightGBM | Hyperparameter tuning | Gridsearch

91
Calculating Root Mean Squared Error (RMSE) with Sklearn and Python
Python

Rmse

86
Dynamically Create Columns in Pandas Dataframe
Python

Pandas | Columns

68
How to Train a Catboost Classifier with GridSearch Hyperparameter Tuning
Python

Catboost | Hyperparameter tuning

68
Overwriting a Python Logging File Instead of Appending
Python

Logging

65