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 vote133 views1 answer
Golang len return type
Go
0 vote126 views7 answers
Https://www.google.com/search?type=guardian
Go
0 vote140 views1 answer
Golang string is digit
Go
0 vote117 views1 answer
Golang set env var
Go
0 vote124 views1 answer
Go for ticker
Go
1 vote133 views1 answer
How to change back to old google playstore console
Go
1 vote132 views1 answer
Append to file in golang
Go
0 vote76 views1 answer
Go convert a string (with decimals) to a floating point number.
Go
0 vote136 views1 answer
2004 argo 6x6 amphibious
Go
1 vote138 views1 answer
Golang jwt example
Go