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 vote250 views1 answer
Convert canvas to DataUrl and insert image in google sheet
Go
0 vote269 views1 answer
Projection on image from camera in godot
Go
0 vote255 views1 answer
Golang read rune from file
Go
0 vote247 views1 answer
Union in go
Go
0 vote251 views1 answer
Run thread golang
Go
0 vote258 views1 answer
How to forward everything from one connection to another in golang
Go
0 vote235 views1 answer
While in golang
Go
1 vote305 views1 answer
Golang parse float64
Go
0 vote267 views2 answers
Golang round float to number of digits precision
Go
0 vote336 views1 answer
Advance web service with golang
Go