What do you want to save?
Add Code snippet
New code examples
-
Other 2022-01-27 05:06:11
pyspark user defined function multiple input
>>> from pyspark.sql.types import IntegerType >>> from pyspark.sql.functions import udf, array >>> sum_cols = udf(lambda arr: sum(arr), IntegerType()) >>> spark.createDataFrame([(101, 1, 16)], ['ID', 'A', 'B']) \ ... ... Add solution