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 vote210 views1 answer
Go remove pacakge
Go
4 votes285 views3 answers
Golang convert string to bytes and convert bytes to string
Go
-1 vote235 views1 answer
How parse table values in golang by table id
Go
7 votes236 views1 answer
Module.exports = router;
Go
0 vote233 views1 answer
Change color google translate select language
Go
0 vote236 views2 answers
Icon material design google fonctionnemnt
Go
5 votes212 views2 answers
Minecraft logo png
Go
0 vote213 views1 answer
Golang http stop redirection
Go
0 vote237 views1 answer
How many days ago was 31 december 1999
Go
0 vote237 views1 answer
Fragment shader using alpha godot
Go