Module mls_lib.data_cleaning.replace_value

Replace Value : Replace Value Data Cleaning Step

Classes

class ReplaceValue (column: str, value_map: dict, data_in: DataFrame)

Replace Value : Replace Value Data Cleaning Step

Expand source code

class ReplaceValue(DataCleaningStep):
    """ Replace Value : Replace Value Data Cleaning Step """
    def __init__(self, column : str, value_map : dict, data_in : DataFrame) -> None:
        super().__init__(
            data_in = data_in
        )
        self.column = column
        self.value_map = value_map

    def execute(self) -> None:
        data = self._get_input("data_in").copy()
        df = data.get_data()
        df[self.column] = df[self.column].map(self.value_map)
        data.set_data(df)
        self._set_output("out", data)

        self._finish_execution()

Ancestors

• DataCleaningStep
• Step

Inherited members

• DataCleaningStep:
  • execute
  • get_output
  • is_finished
  • is_ready