acidplyr.transmute_at

acidplyr.transmute_at(df: DataFrame, vars: list[str], fun: Callable, *args: Any, **kwargs: Any) → DataFrame

Apply a function to specific columns, returning only those columns.

Parameters:

• df (pd.DataFrame) – Input DataFrame.

• vars (list of str) – Column names to transform and return.

• fun (callable) – Function applied to each selected column (as a Series).

• *args – Additional positional arguments passed to fun.

• **kwargs – Additional keyword arguments passed to fun.

Returns:

DataFrame containing only the transformed vars columns.

Return type:

pd.DataFrame

Examples

>>> import pandas as pd
>>> df = pd.DataFrame({"a": [1, 2], "b": [3, 4]})
>>> transmute_at(df, ["a"], lambda s: s * 2).to_dict("list")
{'a': [2, 4]}