Module mls_lib.feature_engineering.join

Join: Component that joins two tables.

Classes

class Join (left: DataFrame, right: DataFrame, how: str, index: list)

Join: Component that joins two tables.

Expand source code

class Join(FeatureEngineeringStep):
    """ Join: Component that joins two tables. """
    def __init__(self, left : DataFrame, right : DataFrame, how : str, index : list) -> None:
        super().__init__(
            left = left,
            right = right,
        )

        self.index = index
        self.how = how

    def execute(self):

        left = self._get_input("left")
        right = self._get_input("right")

        df = left.get_data()
        df = df.join(right.get_data(), on = self.index, how = self.how)

        new_df = DataFrame()
        new_df.set_data(df)
        self._set_output("out", new_df)

        self._finish_execution()

Ancestors

• FeatureEngineeringStep
• Step

Inherited members

• FeatureEngineeringStep:
  • execute
  • get_output
  • is_finished
  • is_ready