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 views2 answers
Get date golang
Go
1 vote154 views2 answers
Golang import alias
Go
5 votes197 views2 answers
Csgo mouse wheel jump bind
Go
0 vote181 views1 answer
How to destroy a gameobject permanently unity when player go through it 2d
Go
8 votes213 views4 answers
Golang string split
Go
-1 vote137 views2 answers
Javascript operators ~~
Go
0 vote183 views1 answer
Golang type
Go
12 votes440 views5 answers
Convert string to int golang
Go
0 vote170 views1 answer
Onion omega location of files on sd
Go
1 vote193 views1 answer
Csgo max ping
Go