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

4 votes236 views4 answers
Is Grepper good?
Go
3 votes173 views4 answers
Variables in golang
Go
0 vote158 views1 answer
Golang url to string
Go
2 votes193 views1 answer
Golang Access-Control-Allow-Origin: '*'
Go
0 vote173 views1 answer
Case in golang
Go
0 vote154 views1 answer
Iterate over iterator golang
Go
0 vote153 views2 answers
On edit copy values to another sheet dynamically google sheets
Go
-1 vote154 views1 answer
Inizialiyze struct di struct golang
Go
0 vote167 views1 answer
Go expand
Go
0 vote158 views2 answers
Go if file exists
Go