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

3 votes121 views1 answer
Big tiddy goth
Go
0 vote131 views1 answer
Go slice pop
Go
0 vote127 views1 answer
Go convert floating point number, to a string, with decimal digits.
Go
1 vote120 views1 answer
Golang read csv file
Go
7 votes134 views2 answers
Golang substring
Go
1 vote126 views1 answer
Golang convert interface to concrete type
Go
0 vote127 views1 answer
How do I disable gogy
Go
1 vote146 views1 answer
Fullscreen switch godot
Go
1 vote112 views2 answers
Golang import alias
Go
0 vote124 views1 answer
How to make history super interesting
Go