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 vote269 views1 answer
Csgo show fps
Go
6 votes265 views1 answer
Never back down
Go
8 votes258 views10 answers
Never gonna give you up
Go
0 vote272 views1 answer
Go array 2 dimensi
Go
0 vote274 views1 answer
Golang http writer redirect
Go
0 vote212 views2 answers
Go slices
Go
2 votes264 views1 answer
Convert float to int golang
Go
0 vote242 views1 answer
Fargo season 4
Go
0 vote253 views1 answer
Googe drive mount google colab
Go
0 vote278 views1 answer
Golang how to print message
Go