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 vote214 views1 answer
Why is nse market going up
Go
0 vote157 views1 answer
If statement string compare go
Go
6 votes228 views1 answer
Stellar meaning
Go
8 votes252 views10 answers
Never gonna give you up
Go
0 vote277 views2 answers
Pies are good
Go
0 vote199 views1 answer
Google sword
Go
0 vote204 views1 answer
Is the ninja airmouse good
Go
0 vote199 views1 answer
Go sum of digits in an integer
Go
0 vote201 views1 answer
Goland print the time it took for code to run
Go
0 vote216 views3 answers
Tushar Jadhav
Go