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 vote217 views4 answers
Buffered channels in golang
Go
0 vote218 views7 answers
If conditional golang
Go
0 vote196 views1 answer
Golang set env var
Go
7 votes222 views1 answer
Golang comp
Go
18 votes447 views2 answers
Connect google drive to colab
Go
3 votes201 views1 answer
Emacs go to line
Go
0 vote186 views1 answer
Upload to shared folder google drive cli linu
Go
0 vote226 views1 answer
First day of the month google sheets
Go
3 votes232 views13 answers
Go hello world
Go
5 votes218 views2 answers
Cs go
Go