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

3 votes118 views1 answer
Go rand seed
Go
0 vote107 views1 answer
Find options mongo golang order by field
Go
9 votes117 views3 answers
Go get from map
Go
0 vote107 views1 answer
Bootstrap remove placeholder on input
Go
1 vote102 views1 answer
Diego maradona
Go
0 vote110 views1 answer
Go circular rotation of array moving right
Go
0 vote97 views1 answer
Go program name
Go
0 vote104 views1 answer
Go in a string, replace multiple spaces with single space
Go
0 vote108 views2 answers
Go Iterating over an array in Golang
Go
2 votes99 views1 answer
Do animals go to heaven
Go