def remove_nan(x): """remove NaN values from data vectors""" imp = Imputer(missing_values='NaN', strategy='mean', axis=0) x_clean = imp.fit_transform(x) return x_clean