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 vote140 views1 answer
Convert canvas to DataUrl and insert image in google sheet
Go
0 vote137 views1 answer
Go array 2 dimensi
Go
1 vote142 views1 answer
Balance teams csgo
Go
0 vote147 views4 answers
Gorm float64
Go
0 vote135 views1 answer
Numpy break number in equal parts
Go
0 vote190 views1 answer
Go type casting
Go
0 vote137 views1 answer
Checking for prime numbers using golang
Go
0 vote228 views1 answer
Advance web service with golang
Go
0 vote146 views1 answer
Golang type declaration
Go
0 vote147 views1 answer
Function change(cash) { // Your code goes here return { two: 0, five: 0, ten: 0 }; }
Go