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 vote365 views1 answer
Check a pair of cell with list of pairs google sheet
Go
0 vote359 views1 answer
Mongodb create schema example
Go
1 vote333 views1 answer
Golang cheat sheet
Go
0 vote399 views1 answer
Ayyubid state. type of government
Go
0 vote352 views1 answer
Convert slice to unique slice golang
Go
0 vote349 views2 answers
Gorm enum
Go
0 vote309 views1 answer
Print struct with keys golang
Go
12 votes419 views5 answers
Golang convert string to int
Go
1 vote314 views1 answer
I cant see script editor google sheet
Go
6 votes311 views6 answers
Use google fonts in css
Go