feat: enhance replace_column to accept a list of new columns

src/safeds/data/tabular/containers/_table.py

@@ -867,9 +867,9 @@ def rename_column(self, old_name: str, new_name: str) -> Table:
         new_df.columns = self._schema.column_names
         return Table._from_pandas_dataframe(new_df.rename(columns={old_name: new_name}))
 
-    def replace_column(self, old_column_name: str, new_column: Column) -> Table:
+    def replace_column(self, old_column_name: str, new_columns: list[Column]) -> Table:
         """
-        Return a copy of the table with the specified old column replaced by a new column. Keeps the order of columns.
+        Return a copy of the table with the specified old column replaced by a list of new columns. Keeps the order of columns.
 
         This table is not modified.
 
@@ -878,44 +878,42 @@ def replace_column(self, old_column_name: str, new_column: Column) -> Table:
         old_column_name : str
             The name of the column to be replaced.
 
-        new_column : Column
-            The new column replacing the old column.
+        new_columns : list[Column]
+            The list of new columns replacing the old column.
 
         Returns
         -------
         result : Table
-            A table with the old column replaced by the new column.
+            A table with the old column replaced by the new columns.
 
         Raises
         ------
         UnknownColumnNameError
             If the old column does not exist.
 
         DuplicateColumnNameError
-            If the new column already exists and the existing column is not affected by the replacement.
+            If at least one of the new columns already exists and the existing column is not affected by the replacement.
 
         ColumnSizeError
-            If the size of the column does not match the amount of rows.
+            If the size of at least one of the new columns does not match the amount of rows.
         """
         if old_column_name not in self._schema.column_names:
             raise UnknownColumnNameError([old_column_name])
 
-        if new_column.name in self._schema.column_names and new_column.name != old_column_name:
-            raise DuplicateColumnNameError(new_column.name)
-
-        if self.number_of_rows != new_column._data.size:
-            raise ColumnSizeError(str(self.number_of_rows), str(new_column._data.size))
+        columns = list[Column]()
+        for old_column in self.schema.column_names:
+            if old_column == old_column_name:
+                for new_column in new_columns:
+                    if new_column.name in self._schema.column_names and new_column.name != old_column_name:
+                        raise DuplicateColumnNameError(new_column.name)
 
-        if old_column_name != new_column.name:
-            renamed_table = self.rename_column(old_column_name, new_column.name)
-            result = renamed_table._data
-            result.columns = renamed_table._schema.column_names
-        else:
-            result = self._data.copy()
-            result.columns = self._schema.column_names
+                    if self.number_of_rows != new_column.number_of_rows:
+                        raise ColumnSizeError(str(self.number_of_rows), str(new_column._data.size))
+                    columns.append(new_column)
+            else:
+                columns.append(self.get_column(old_column))
 
-        result[new_column.name] = new_column._data
-        return Table._from_pandas_dataframe(result)
+        return Table.from_columns(columns)
 
     def shuffle_rows(self) -> Table:
         """
@@ -1103,7 +1101,7 @@ def transform_column(self, name: str, transformer: Callable[[Row], Any]) -> Tabl
         """
         if self.has_column(name):
             items: list = [transformer(item) for item in self.to_rows()]
-            result: Column = Column(name, items)
+            result: list[Column] = [Column(name, items)]
             return self.replace_column(name, result)
         raise UnknownColumnNameError([name])
 

tests/safeds/data/tabular/containers/_table/test_replace_column.py

@@ -8,7 +8,7 @@
 
 
 @pytest.mark.parametrize(
-    ("table", "column_name", "column", "expected"),
+    ("table", "column_name", "columns", "expected"),
     [
         (
             Table(
@@ -18,13 +18,14 @@
                     "C": ["a", "b", "c"],
                 },
             ),
-            "C",
-            Column("C", ["d", "e", "f"]),
+            "B",
+            [Column("B", ["d", "e", "f"]), Column("D", [3, 4, 5])],
             Table(
                 {
                     "A": [1, 2, 3],
-                    "B": [4, 5, 6],
-                    "C": ["d", "e", "f"],
+                    "B": ["d", "e", "f"],
+                    "D": [3, 4, 5],
+                    "C": ["a", "b", "c"],
                 },
             ),
         ),
@@ -37,7 +38,7 @@
                 },
             ),
             "C",
-            Column("D", ["d", "e", "f"]),
+            [Column("D", ["d", "e", "f"])],
             Table(
                 {
                     "A": [1, 2, 3],
@@ -48,8 +49,8 @@
         ),
     ],
 )
-def test_should_replace_column(table: Table, column_name: str, column: Column, expected: Table) -> None:
-    result = table.replace_column(column_name, column)
+def test_should_replace_column(table: Table, column_name: str, columns: list[Column], expected: Table) -> None:
+    result = table.replace_column(column_name, columns)
     assert result == expected
 
 
@@ -75,7 +76,7 @@ def test_should_raise_error(
             "C": ["a", "b", "c"],
         },
     )
-    column = Column(column_name, column_values)
+    column = [Column(column_name, column_values)]
 
     with pytest.raises(error):
         input_table.replace_column(old_column_name, column)