simple imputer for categorical data pandas code example Example: how to store categorical variables in separate dataframe df.loc[:,df.dtypes==np.object]