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 vote194 views1 answer
Anaconda google colab
Go
3 votes188 views1 answer
Go rand seed
Go
10 votes154 views1 answer
How to do google search
Go
8 votes208 views4 answers
Strings split golang
Go
0 vote130 views1 answer
GO GO ANIEM
Go
1 vote163 views1 answer
Include relation in mongo query
Go
0 vote192 views1 answer
Prevent mac to go back on scroll
Go
0 vote172 views1 answer
Golang interface pointer receiver
Go
1 vote173 views1 answer
Failed to execute goal org.apache.maven.plugins:maven-surefire-plugin:2.12.4:test (default-test) on project mockito-course: There are test failures.
Go
0 vote173 views1 answer
How to make history super interesting
Go