snowflake.snowpark.functions.call_udf¶

snowflake.snowpark.functions.call_udf(udf_name: str, *args: Union[Column, None, bool, int, float, str, bytearray, Decimal, date, datetime, time, bytes, NaTType, float64, list, tuple, dict]) → Column[source] (https://github.com/snowflakedb/snowpark-python/blob/v1.16.0/src/snowflake/snowpark/functions.py#L8046-L8073)¶

Calls a user-defined function (UDF) by name.

Parameters:

udf_name – The name of UDF in Snowflake.
args –
Arguments can be in two types:
- Column, or
- Basic Python types, which are converted to Snowpark literals.

Example::

>>> from snowflake.snowpark.types import IntegerType
>>> udf_def = session.udf.register(lambda x, y: x + y, name="add_columns", input_types=[IntegerType(), IntegerType()], return_type=IntegerType(), replace=True)
>>> df = session.create_dataframe([[1, 2]], schema=["a", "b"])
>>> df.select(call_udf("add_columns", col("a"), col("b"))).show()
-------------------------------
|"ADD_COLUMNS(""A"", ""B"")"  |
-------------------------------
|3                            |
-------------------------------

Copy