@pandas_udf(returnType=as_spark_type(pser.dtype.type))
def calculate_columns_axis(*cols: pd.Series) -> pd.Series:
    return getattr(pd.concat(cols, axis=1), name)(axis=axis, numeric_only=numeric_only, **kwargs)
@pandas_udf(returnType=as_spark_type(pser.dtype.type))
def calculate_columns_axis(*cols: pd.Series) -> pd.Series:
    return getattr(pd.concat(cols, axis=1), name)(axis=axis, numeric_only=numeric_only, **kwargs)