Impute attributes from a dataframe

attributes_imputer.py

def imputeMissingValues(dataframe, desired_strategy, attributes_to_impute):
    from sklearn.preprocessing import Imputer

    for attr in attributes_to_impute:
        values_ = dataframe[attr].values.reshape(-1, 1)
        imp = Imputer(missing_values=np.nan, strategy=desired_strategy, axis=0)
        imp.fit(values_)
        transformed_values = imp.transform(values_)
        dataframe.loc[:,attr] = transformed_values
    return dataframe