Pyspark udf multiple inputs

from pyspark.sql.types import IntegerType
from pyspark.sql.functions import udf

def sum(x, y):
    return x + y

sum_cols = udf(sum, IntegerType())

a=spark.createDataFrame([(101, 1, 16)], ['ID', 'A', 'B'])
a.withColumn('Result', sum_cols('A', 'B')).show()

Source: stackoverflow.com

Similar codes

0 vote337 views1 answer
Custom echo golang middleware
Go
9 votes334 views3 answers
Go add to map
Go
1 vote309 views1 answer
What is the update function in godot?
Go
16 votes416 views6 answers
Never gonna give you up release date
Go
0 vote362 views1 answer
Golang eliminate duplicate spaces
Go
4 votes364 views4 answers
Is Grepper good?
Go
5 votes282 views2 answers
Interface to string golang
Go
0 vote351 views1 answer
Go error handling
Go
1 vote318 views1 answer
Search google at your own risk
Go
0 vote288 views1 answer
How to make sure that google colab runtime is not expired
Go