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 vote449 views1 answer
Golang send email jordan-wright
Go
0 vote384 views1 answer
Printf ("%d \n", goal);
Go
0 vote384 views1 answer
Google calendar remove a goal
Go
0 vote437 views1 answer
Aws s3 get file size
Go
0 vote408 views1 answer
Golang truncate float to 1 decimal place
Go
0 vote390 views1 answer
Why was dual government abolished in bengal
Go
0 vote405 views1 answer
Type golang
Go
16 votes615 views6 answers
Golang map
Go
1 vote419 views1 answer
Real time currency conversion google sheets
Go
0 vote396 views1 answer
Golang recursive function example
Go