feat: with_column(s)_renamed expression for DataFrame

[jessie-young]

issue #3705

Add with_column_renamed and with_columns_renamed to DataFrame API (no SQL).

def with_column_renamed(self, existing: str, new: str) -> "DataFrame":
    """Renames a column in the current DataFrame.

    Example:
        >>> import daft
        >>> df = daft.from_pydict({"x": [1, 2, 3], "y": [4, 5, 6]})
        >>> df.with_column_renamed("x", "foo").show()
        ╭───────┬───────╮
        │ foo   ┆ y     │
        │ ---   ┆ ---   │
        │ Int64 ┆ Int64 │
        ╞═══════╪═══════╡
        │ 1     ┆ 4     │
        ├╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌┤
        │ 2     ┆ 5     │
        ├╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌┤
        │ 3     ┆ 6     │
        ╰───────┴───────╯
        <BLANKLINE>
        (Showing first 3 of 3 rows)

    Args:
        existing (str): name of the existing column to rename
        new (str): new name for the column

    Returns:
        DataFrame: DataFrame with the column renamed.
    """
    
def with_columns_renamed(self, colsMap: Dict[str, str]) -> "DataFrame":
    """Renames multiple columns in the current DataFrame.

    Example:
        >>> import daft
        >>> df = daft.from_pydict({"x": [1, 2, 3], "y": [4, 5, 6]})
        >>> df.with_columns_renamed({"x": "foo", "y": "bar"}).show()
        ╭───────┬───────╮
        │ foo   ┆ bar   │
        │ ---   ┆ ---   │
        │ Int64 ┆ Int64 │
        ╞═══════╪═══════╡
        │ 1     ┆ 4     │
        ├╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌┤
        │ 2     ┆ 5     │
        ├╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌┤
        │ 3     ┆ 6     │
        ╰───────┴───────╯
        <BLANKLINE>
        (Showing first 3 of 3 rows)
    """

[codspeed-hq]

CodSpeed Performance Report

Merging #3732 will not alter performance

_{Comparing with-columns_renamed (10aa453) with main (fec399e)}

Summary

✅ 27 untouched benchmarks

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 77.59%. Comparing base (fec399e) to head (10aa453).
Report is 2 commits behind head on main.

Additional details and impacted files

@@           Coverage Diff           @@
##             main    #3732   +/-   ##
=======================================
  Coverage   77.58%   77.59%           
=======================================
  Files         729      729           
  Lines       92010    92042   +32     
=======================================
+ Hits        71389    71421   +32     
  Misses      20621    20621           

Files with missing lines	Coverage Δ
daft/dataframe/dataframe.py	85.51% <100.00%> (+0.11%)	⬆️
daft/logical/builder.py	90.65% <100.00%> (+0.37%)	⬆️
src/daft-logical-plan/src/builder/mod.rs	91.81% <100.00%> (+0.15%)	⬆️

[colin-ho]

Awesome, LGTM!