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 vote135 views1 answer
Lowwer in go
Go
0 vote129 views1 answer
Gorm models
Go
0 vote132 views1 answer
Checking for prime numbers using golang
Go
0 vote148 views1 answer
How to open a file in gofer
Go
19 votes267 views6 answers
Mongo delete all documents
Go
0 vote91 views1 answer
How to upload data to google colab
Go
2 votes147 views2 answers
Go iterate over slice
Go
0 vote140 views1 answer
Go value of pi
Go
0 vote143 views1 answer
Explosion godot
Go
0 vote141 views1 answer
Go select timeout
Go