99095d5412483ec623278bdb1a0c9e24b18bfc85,pdpipe/basic_stages.py,MapColVals,_op,#MapColVals#Any#Any#,497

Before Change

    def _op(self, df, verbose):
        inter_df = df
        source_column = df[self._column_name]
        if self._drop:
            inter_df = df.drop(self._column_name, axis=1)
        inter_df = df.assign(**{
            self._result_col_name: source_column.replace(self._value_map)
        })
        return inter_df

    def __str__(self):
        return "Map values of column {}.".format(self._column_name)

After Change

    def _op(self, df, verbose):
        inter_df = df
        for i, colname in enumerate(self._columns):
            source_col = df[colname]
            loc = df.columns.get_loc(colname) + 1
            new_name = self._result_columns[i]
            if self._drop:
                inter_df = inter_df.drop(colname, axis=1)
                loc -= 1
            inter_df = out_of_place_col_insert(
                df=inter_df,
                series=source_col.map(self._value_map),
                loc=loc,
                column_name=new_name)
        return inter_df


class Encode(PipelineStage):

Instances