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 vote404 views1 answer
Check if directory exists in go
Go
3 votes395 views1 answer
How to join google
Go
0 vote339 views1 answer
Why is my discord bot keep going offline
Go
0 vote429 views1 answer
Roblox lua how to make something go invisible
Go
0 vote383 views10 answers
Ljljkl
Go
0 vote352 views1 answer
How to make a div text don't go to the new line
Go
0 vote368 views2 answers
Print unicode character in golang
Go
0 vote349 views1 answer
Go Pause execution for several seconds
Go
0 vote351 views1 answer
Find options mongo golang order by field
Go
0 vote355 views1 answer
Go jump back to an outer loop
Go