Esempio n. 1
0
def drop_feature(fn, feature, fn_out=None):
    Filefunctions.exists_assert(fn)
    assert (isinstance(feature, str) or isinstance(feature, int))
    df = import_df(fn)
    if isinstance(feature, int):
        feature = df.columns[feature]

    if feature not in df.columns:
        warn('Feature is not a column')
        return

    df.drop(labels=[feature], axis=1, inplace=True)
    if fn_out is None:
        fn_out = fn
    export_df(df, fn_out)
Esempio n. 2
0
def sort_file_on_date(fn_in, feature, fn_out=None):
    assert (isinstance(feature, str) or isinstance(
        feature, int)), 'Feature must be str or int: {}'.format(feature)
    Filefunctions.exists_assert(fn_in)
    df = import_df(fn_in)

    if isinstance(feature, int):
        feature = df.columns[feature]

    fixDates(frame=df, date_column_string=feature)

    df.sort_values(by=[feature], ascending=True, inplace=True)

    if fn_out is None:
        fn_out = fn_in

    export_df(df, fn_out)