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 vote133 views1 answer
How to get data in nested map go
Go
4 votes115 views2 answers
Appending map into map golang
Go
1 vote122 views1 answer
Golang jwt
Go
0 vote121 views2 answers
How to parse io.nopCloser golang
Go
8 votes167 views1 answer
Golang delete element from array
Go
0 vote116 views1 answer
Recenter google map based on route
Go
0 vote109 views1 answer
Go nested slice
Go
0 vote123 views1 answer
Invalid set index godot
Go
5 votes122 views1 answer
Initialize map in golang
Go
0 vote68 views1 answer
Kubernetes field selector
Go