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 votes119 views2 answers
Uint64 to byte golang
Go
0 vote103 views1 answer
Add text with marker google map android
Go
0 vote104 views1 answer
Stop by interrupt
Go
0 vote118 views1 answer
Visual studio code you are neither in a module nor in your gopath
Go
0 vote117 views1 answer
Go create multiple variables on a single line
Go
0 vote113 views1 answer
2004 argo 6x6 amphibious
Go
1 vote106 views1 answer
Wordpress add google fonts
Go
0 vote106 views1 answer
Golang find nearest number to array
Go
7 votes106 views1 answer
TypeError: Router.use() requires a middleware function but got a Object
Go
0 vote96 views1 answer
Media google planilhas
Go