categorical pandas dataframe code example

Example 1: 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"]

Example 2: how to store categorical variables in separate dataframe

df.loc[:,df.dtypes==np.object]