516b77610d058cfe3a6f379e8ff9753065d48707,janitor/functions.py,,groupby_agg,#,3208

Before Change

    :returns: A pandas DataFrame.
    
    new_col = df.groupby(by)[agg_column_name].transform(agg)
    df_new = df.assign(**{new_column_name: new_col})
    return df_new


@pf.register_dataframe_accessor("data_description")

After Change

    // .isnull().any()
    if any(df[col].hasnans for col in by):

        mapping = {
            column: ".*^%s1ho1go1logoban?*&-|/\\gos1he()//_" for column in by
        }

        df[new_column_name] = (
            df.fillna(mapping).groupby(by)[agg_column_name].transform(agg)
        )

Instances