Imputing Categorical Variable Using Python Machine Learning