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
Gohugo windows
Go
0 vote137 views1 answer
Set in go
Go
0 vote122 views1 answer
Fargo season 4
Go
1 vote127 views1 answer
Golang echo print
Go
0 vote135 views1 answer
When do we need & in golang
Go
0 vote146 views1 answer
How to open a file in gofer
Go
0 vote138 views1 answer
Golang bufio gjson
Go
0 vote145 views1 answer
Go remove pacakge
Go
27 votes267 views4 answers
Updates were rejected because the tip of your current branch is behind
Go
0 vote131 views1 answer
Golang execute bash command
Go