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 vote252 views1 answer
Oauth with google and vue
Go
-1 vote544 views1 answer
Welcome Firebase Hosting Setup Complete You're seeing this because you've successfully setup Firebase Hosting. Now it's time to go build something extraordinary! OPEN HOSTING DOCUMENTATION
Go
1 vote243 views3 answers
Interface to int golang
Go
2 votes230 views1 answer
Reverse a string in golang
Go
3 votes226 views1 answer
Discord golang
Go
0 vote290 views1 answer
Create json in go
Go
2 votes235 views1 answer
How good test case should be
Go
1 vote252 views3 answers
Golang generate random string
Go
1 vote221 views1 answer
I cant see script editor google sheet
Go
0 vote223 views1 answer
How do I disable gogy
Go