4e1bb8d51e8b01c22cf52a0a5a8d7466083dd2da,category_encoders/polynomial.py,PolynomialEncoder,polynomial_coding,#Any#Any#,238

Before Change

        for switch in mapping:
            col = switch.get("col")
            mod = switch.get("mapping")
            new_columns = []
            for i in range(len(mod.columns)):
                c = mod.columns[i]
                new_col = str(col) + "_%d" % (i, )
                X.loc[:, new_col] = mod[c].loc[X[col]].values
                new_columns.append(new_col)
            old_column_index = cols.index(col)
            cols[old_column_index: old_column_index + 1] = new_columns

        cols = ["intercept"] + cols
        X = X.reindex(columns=cols)

        return X

After Change

        
        

        X = X_in.copy(deep=True)

        cols = X.columns.values.tolist()

        X["intercept"] = pd.Series([1] * X.shape[0], index=X.index)

        for switch in mapping:
            col = switch.get("col")
            mod = switch.get("mapping")

            base_df = mod.loc[X[col]]
            base_df.set_index(X.index, inplace=True)
            X = pd.concat([base_df, X], axis=1)

            old_column_index = cols.index(col)
            cols[old_column_index: old_column_index + 1] = mod.columns

        cols = ["intercept"] + cols

        return X.reindex(columns=cols)

Instances