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 vote201 views1 answer
Golang reader from byte array
Go
0 vote201 views1 answer
Run-time reflection means in go
Go
0 vote193 views1 answer
Recenter google map based on route
Go
0 vote207 views1 answer
Aws s3 rm
Go
1 vote212 views1 answer
Real time currency conversion google sheets
Go
0 vote187 views1 answer
Visual studio code you are neither in a module nor in your gopath
Go
1 vote171 views2 answers
Golang import alias
Go
-1 vote172 views1 answer
Cast interface to struct golang
Go
0 vote186 views1 answer
Package is not in GOROOT (/usr/local/go/src/) - golang
Go
2 votes206 views1 answer
How to reload current scene godot
Go