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

8 votes157 views4 answers
Golang string split
Go
-2 votes139 views1 answer
How to abandon changes and go last commit
Go
5 votes141 views2 answers
Minecraft logo png
Go
0 vote139 views1 answer
Salutem ex gonsas
Go
0 vote140 views1 answer
Golang switch no condition
Go
-1 vote134 views1 answer
Cast interface to struct golang
Go
0 vote145 views2 answers
Golang Insert At Index (any slice)
Go
0 vote125 views1 answer
Golang truncate float to 1 decimal place
Go
0 vote138 views2 answers
Golang default value
Go
0 vote153 views2 answers
Get date golang
Go