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 vote371 views1 answer
Golang maximum length for numbers
Go
0 vote267 views1 answer
Golang decorator
Go
0 vote280 views1 answer
Gorm example onerealation golang
Go
0 vote252 views1 answer
Golang type enum
Go
0 vote293 views1 answer
Go create multiple variables on a single line
Go
1 vote279 views3 answers
Interface to int golang
Go
0 vote271 views1 answer
Gospider
Go
0 vote325 views1 answer
Duck duck go retro search widget
Go
-1 vote251 views1 answer
Go functions
Go
0 vote293 views1 answer
Select a random number between 1 and 5 in golang
Go