dataset categorical variables code example Example: how to store categorical variables in separate dataframe df.loc[:,df.dtypes==np.object]