Core concept

@pandas_udf("double")
def my_udf(v: pd.Series) -> pd.Series:
    return v * 2
Advertisement

How it works

Requires PyArrow. Batch size tuning.

Advertisement

Trade-offs + gotchas

Python UDFs on PySpark.