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 vote331 views1 answer
Pymongo timeout
Go
0 vote368 views1 answer
Excelize
Go
0 vote316 views1 answer
How to declare a float in golang
Go
1 vote394 views1 answer
Nasi goreng”
Go
0 vote346 views1 answer
Download file in go
Go
0 vote325 views1 answer
Golang len return type
Go
4 votes323 views2 answers
Appending map into map golang
Go
6 votes356 views1 answer
Never back down
Go
0 vote331 views1 answer
Golang before all tests
Go
0 vote371 views1 answer
Golang default keyword
Go