from sklearn.impute import SimpleImputer imputer = SimpleImputer(strategy="median") imputed = imputer.fit_transform(numerical_df) numerical_df_imputed = pd.DataFrame(imputed, columns=numerical_df.columns, index=numerical_df.index) numerical_df_imputed