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

5 votes66 views3 answers
Go for loop slice
Go
27 votes257 views4 answers
Updates were rejected because the tip of your current branch is behind
Go
0 vote132 views1 answer
Golang url to string
Go
0 vote126 views1 answer
Gorm example onerealation golang
Go
0 vote156 views1 answer
Golang how to print message
Go
0 vote122 views1 answer
How to get your name at top of google search
Go
0 vote129 views1 answer
How many days ago was 31 december 1999
Go
0 vote193 views1 answer
Go to a specific header in a website
Go
2 votes125 views1 answer
How to create custom google home commands
Go
0 vote123 views1 answer
Iranrepo
Go