Source code for libpyhat.transform.scale

from sklearn.preprocessing import StandardScaler




def do_scale(df_to_scale, df_to_fit=None, spect_label="wvl"):
    # If no data frame is provided to fit, then fit the scaler on the same
    # data frame to be scaled
    if df_to_fit is None:
        df_to_fit = df_to_scale

    scaler = StandardScaler()
    scaler.fit(df_to_fit[spect_label])
    df_to_scale[spect_label] = scaler.transform(df_to_scale[spect_label])

    return df_to_scale, scaler