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 vote385 views1 answer
Render_to_response() got an unexpected keyword argument 'context_to_response'
Go
0 vote379 views1 answer
How to destroy a gameobject permanently unity when player go through it 2d
Go
0 vote388 views1 answer
Golang find nearest number to array
Go
0 vote431 views1 answer
How many days ago was 31 december 1999
Go
0 vote392 views1 answer
Golang decorator
Go
0 vote426 views10 answers
Ljljkl
Go
0 vote400 views1 answer
Using either macro within goto sugarcube
Go
1 vote381 views2 answers
Golang import alias
Go
0 vote409 views1 answer
Airteltigo money short code
Go
0 vote431 views1 answer
Golang bufio gjson
Go