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

1 vote341 views2 answers
Constants in golang
Go
0 vote371 views1 answer
How to xss google form
Go
0 vote347 views1 answer
Go Shorthand Declaration of Array in Golang
Go
0 vote429 views1 answer
String find index in golang
Go
0 vote337 views1 answer
Go array 2 dimensi
Go
0 vote307 views6 answers
The go blog defer panic recover
Go
0 vote364 views1 answer
Vscode golang cannot find package
Go
0 vote306 views1 answer
Goroutines in golang
Go
0 vote330 views1 answer
Gorm array type data postgres
Go
0 vote313 views1 answer
Gorm models
Go