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

2 votes150 views1 answer
Convert float to int golang
Go
2 votes144 views1 answer
Golang test coverage
Go
2 votes140 views1 answer
How to reload current scene godot
Go
1 vote182 views1 answer
Add google font to wordpress theme
Go
0 vote150 views1 answer
Get data from pymongo without id
Go
8 votes235 views4 answers
Golang []byte to string
Go
0 vote135 views1 answer
Golang scan directory
Go
0 vote134 views1 answer
Go nested slice
Go
1 vote149 views1 answer
Go fmt all files
Go
4 votes174 views1 answer
Go remove from map
Go