siuba icon indicating copy to clipboard operation
siuba copied to clipboard

consolidate when dispatching DataFrame function for a DataFrameGroupBy

Open machow opened this issue 4 years ago • 0 comments

in siuba.dply.verbs, lots of this repeated right now...

@unite.register(DataFrameGroupBy)
def _unite_gdf(__data, *args, **kwargs):
    # TODO: consolidate these trivial group by dispatched funcs

    groupings = __data.grouper.groupings

    df = __data.obj

    f_unite = unite.registry[pd.DataFrame]
    out = f_unite(df, *args, **kwargs)

    return out.groupby(groupings)

machow avatar Aug 18 '19 14:08 machow